apunkt/MetaGPT
1
0
Fork 0
mirror of https://github.com/FoundationAgents/MetaGPT.git synced 2026-05-24 14:15:17 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

update dataset.py

Browse source
This commit is contained in:
Yizhou Chi 2024-09-09 13:54:09 +08:00
parent 93ff1f8f2b
commit 401ca97846
1 changed files with 6 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

12
expo/data/dataset.py
View file

@ -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)