change label column

This commit is contained in:
Yizhou Chi 2024-09-27 15:13:23 +08:00
parent 1654174b15
commit af844693b1
2 changed files with 3 additions and 3 deletions

View file

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

View file

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