diff --git a/expo/data/hf_data.py b/expo/data/hf_data.py index df3a6ed20..a43fcd415 100644 --- a/expo/data/hf_data.py +++ b/expo/data/hf_data.py @@ -18,7 +18,7 @@ HFDATSETS = [ "name": "oxford-iiit-pet", "dataset_name": "timm/oxford-iiit-pet", "image_col": "image", - "target_col": "label_cat_dog", + "target_col": "label", "modality": "image", }, { @@ -115,7 +115,7 @@ class HFExpDataset(ExpDataset): if __name__ == "__main__": dataset_dir = "D:/work/automl/datasets" - save_analysis_pool = False + save_analysis_pool = True force_update = False datasets_dict = {"datasets": {}} solution_designer = SolutionDesigner() diff --git a/expo/datasets.yaml b/expo/datasets.yaml index 92e004c6d..e58e717b5 100644 --- a/expo/datasets.yaml +++ b/expo/datasets.yaml @@ -202,7 +202,7 @@ datasets: oxford-iiit-pet: dataset: oxford-iiit-pet metric: f1 - target_col: label_cat_dog + target_col: label user_requirement: "This is a oxford-iiit-pet dataset. Your goal is to predict\ \ the target column `label_cat_dog`.\nPerform data analysis, data preprocessing,\ \ feature engineering, and modeling to predict the target. \nReport f1 on the\