MetaGPT/expo/data.yaml

datasets_dir: "D:/work/automl/datasets" # path to the datasets directory

datasets:
  titanic:
    dataset: 04_titanic
    metric: f1
    user_requirement: "This is a 04_titanic dataset. Your goal is to predict the target\
      \ column `Survived`.\nPerform data analysis, data preprocessing, feature engineering,\
      \ and modeling to predict the target. \nReport f1 on the eval data. Do not plot\
      \ or make any visualizations.\n"

  house_prices:
    dataset: 05_house-prices-advanced-regression-techniques
    metric: rmse
    user_requirement: "This is a 05_house-prices-advanced-regression-techniques dataset.\
      \ Your goal is to predict the target column `SalePrice`.\nPerform data analysis,\
      \ data preprocessing, feature engineering, and modeling to predict the target.\
      \ \nReport rmse on the eval data. Do not plot or make any visualizations.\n"

  santander_customers:
    dataset: 06_santander-customer-transaction-prediction
    metric: f1
    user_requirement: "This is a 06_santander-customer-transaction-prediction dataset.\
      \ Your goal is to predict the target column `target`.\nPerform data analysis,\
      \ data preprocessing, feature engineering, and modeling to predict the target.\
      \ \nReport f1 on the eval data. Do not plot or make any visualizations.\n"
  icr:
    dataset: 07_icr-identify-age-related-conditions
    metric: f1
    user_requirement: "This is a 07_icr-identify-age-related-conditions dataset. Your\
      \ goal is to predict the target column `Class`.\nPerform data analysis, data\
      \ preprocessing, feature engineering, and modeling to predict the target. \n\
      Report f1 on the eval data. Do not plot or make any visualizations.\n"

  lick_prediction_small:
    dataset: Click_prediction_small
    metric: f1
    user_requirement: "This is a Click_prediction_small dataset. Your goal is to predict\
      \ the target column `click`.\nPerform data analysis, data preprocessing, feature\
      \ engineering, and modeling to predict the target. \nReport f1 on the eval data.\
      \ Do not plot or make any visualizations.\n"

  GesturePhaseSegmentationProcessed:
    dataset: GesturePhaseSegmentationProcessed
    metric: f1 weighted
    user_requirement: "This is a GesturePhaseSegmentationProcessed dataset. Your goal\
      \ is to predict the target column `Phase`.\nPerform data analysis, data preprocessing,\
      \ feature engineering, and modeling to predict the target. \nReport f1 weighted\
      \ on the eval data. Do not plot or make any visualizations.\n"

  Moneyball:
    dataset: Moneyball
    metric: rmse
    user_requirement: "This is a Moneyball dataset. Your goal is to predict the target\
      \ column `RS`.\nPerform data analysis, data preprocessing, feature engineering,\
      \ and modeling to predict the target. \nReport rmse on the eval data. Do not\
      \ plot or make any visualizations.\n"

  SAT11-HAND-runtime-regression:
    dataset: SAT11-HAND-runtime-regression
    metric: rmse
    user_requirement: "This is a SAT11-HAND-runtime-regression dataset. Your goal\
      \ is to predict the target column `runtime`.\nPerform data analysis, data preprocessing,\
      \ feature engineering, and modeling to predict the target. \nReport rmse on\
      \ the eval data. Do not plot or make any visualizations.\n"

  boston:
    dataset: boston
    metric: rmse
    user_requirement: "This is a boston dataset. Your goal is to predict the target\
      \ column `MEDV`.\nPerform data analysis, data preprocessing, feature engineering,\
      \ and modeling to predict the target. \nReport rmse on the eval data. Do not\
      \ plot or make any visualizations.\n"

  colleges:
    dataset: colleges
    metric: rmse
    user_requirement: "This is a colleges dataset. Your goal is to predict the target\
      \ column `percent_pell_grant`.\nPerform data analysis, data preprocessing, feature\
      \ engineering, and modeling to predict the target. \nReport rmse on the eval\
      \ data. Do not plot or make any visualizations.\n"

  credit-g:
    dataset: credit-g
    metric: f1
    user_requirement: "This is a credit-g dataset. Your goal is to predict the target\
      \ column `class`.\nPerform data analysis, data preprocessing, feature engineering,\
      \ and modeling to predict the target. \nReport f1 on the eval data. Do not plot\
      \ or make any visualizations.\n"

  diamonds:
    dataset: diamonds
    metric: rmse
    user_requirement: "This is a diamonds dataset. Your goal is to predict the target\
      \ column `price`.\nPerform data analysis, data preprocessing, feature engineering,\
      \ and modeling to predict the target. \nReport rmse on the eval data. Do not\
      \ plot or make any visualizations.\n"

  jasmine:
    dataset: jasmine
    metric: f1
    user_requirement: "This is a jasmine dataset. Your goal is to predict the target\
      \ column `class`.\nPerform data analysis, data preprocessing, feature engineering,\
      \ and modeling to predict the target. \nReport f1 on the eval data. Do not plot\
      \ or make any visualizations.\n"

  kc1:
    dataset: kc1
    metric: f1
    user_requirement: "This is a kc1 dataset. Your goal is to predict the target column\
      \ `defects`.\nPerform data analysis, data preprocessing, feature engineering,\
      \ and modeling to predict the target. \nReport f1 on the eval data. Do not plot\
      \ or make any visualizations.\n"

  kick:
    dataset: kick
    metric: f1
    user_requirement: "This is a kick dataset. Your goal is to predict the target\
      \ column `IsBadBuy`.\nPerform data analysis, data preprocessing, feature engineering,\
      \ and modeling to predict the target. \nReport f1 on the eval data. Do not plot\
      \ or make any visualizations.\n"

  mfeat-factors:
    dataset: mfeat-factors
    metric: f1 weighted
    user_requirement: "This is a mfeat-factors dataset. Your goal is to predict the\
      \ target column `class`.\nPerform data analysis, data preprocessing, feature\
      \ engineering, and modeling to predict the target. \nReport f1 weighted on the\
      \ eval data. Do not plot or make any visualizations.\n"

  segment:
    dataset: segment
    metric: f1 weighted
    user_requirement: "This is a segment dataset. Your goal is to predict the target\
      \ column `class`.\nPerform data analysis, data preprocessing, feature engineering,\
      \ and modeling to predict the target. \nReport f1 weighted on the eval data.\
      \ Do not plot or make any visualizations.\n"

  steel-plates-fault:
    dataset: steel-plates-fault
    metric: f1 weighted
    user_requirement: "This is a steel-plates-fault dataset. Your goal is to predict\
      \ the target column `target`.\nPerform data analysis, data preprocessing, feature\
      \ engineering, and modeling to predict the target. \nReport f1 weighted on the\
      \ eval data. Do not plot or make any visualizations.\n"

  wine-quality-white:
    dataset: wine-quality-white
    metric: f1 weighted
    user_requirement: "This is a wine-quality-white dataset. Your goal is to predict\
      \ the target column `Class`.\nPerform data analysis, data preprocessing, feature\
      \ engineering, and modeling to predict the target. \nReport f1 weighted on the\
      \ eval data. Do not plot or make any visualizations.\n"


work_dir: ../workspace # path to the workspace directory
role_dir: storage/team/environment/roles/ResearchAssistant_David
# analysis_pool_dir: D:/work/MG-open/MetaGPT/examples/MCTS_test/analysis_pool_sample.json