From 401ca97846b91a97629c4a8144b29c44d79fe1bb Mon Sep 17 00:00:00 2001 From: Yizhou Chi Date: Mon, 9 Sep 2024 13:54:09 +0800 Subject: [PATCH] update dataset.py --- expo/data/dataset.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/expo/data/dataset.py b/expo/data/dataset.py index 4ac931d9f..8bcce0b1a 100644 --- a/expo/data/dataset.py +++ b/expo/data/dataset.py @@ -338,13 +338,13 @@ if __name__ == "__main__": save_analysis_pool = True datasets_dict = {"datasets": {}} solution_designer = SolutionDesigner() - # for dataset_id in OPENML_DATASET_IDS: - # openml_dataset = OpenMLExpDataset("", datasets_dir, dataset_id, force_update=force_update) - # asyncio.run(process_dataset(openml_dataset, solution_designer, save_analysis_pool, datasets_dict)) + for dataset_id in OPENML_DATASET_IDS: + openml_dataset = OpenMLExpDataset("", datasets_dir, dataset_id, force_update=force_update) + asyncio.run(process_dataset(openml_dataset, solution_designer, save_analysis_pool, datasets_dict)) - # for dataset_name, target_col in CUSTOM_DATASETS: - # custom_dataset = ExpDataset(dataset_name, datasets_dir, target_col=target_col, force_update=force_update) - # asyncio.run(process_dataset(custom_dataset, solution_designer, save_analysis_pool, datasets_dict)) + for dataset_name, target_col in CUSTOM_DATASETS: + custom_dataset = ExpDataset(dataset_name, datasets_dir, target_col=target_col, force_update=force_update) + asyncio.run(process_dataset(custom_dataset, solution_designer, save_analysis_pool, datasets_dict)) for dataset_name, target_col in DSAGENT_DATASETS: custom_dataset = ExpDataset(dataset_name, datasets_dir, target_col=target_col, force_update=force_update)