mirror of
https://github.com/FoundationAgents/MetaGPT.git
synced 2026-06-11 15:15:18 +02:00
change label column
This commit is contained in:
parent
1654174b15
commit
af844693b1
2 changed files with 3 additions and 3 deletions
|
|
@ -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()
|
||||
|
|
|
|||
|
|
@ -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\
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue