MetaGPT/expo/README.md
2024-09-02 20:23:45 +08:00

1.3 KiB
Raw Blame History

Expo

Setup

In the root directory, pip install -e .

cd expo

pip install -r requirements.txt

Instruction

Examples

Run Base DI

python run_experiment.py --exp_mode base --task titanic

Run DI RandExp

  • Single insight python run_experiment.py --exp_mode aug --task titanic --aug_mode single

  • Set insight python run_experiment.py --exp_mode aug --task titanic --aug_mode set

Run DI MCTS

python run_experiment.py --exp_mode mcts --task titanic --rollout 5

If the dataset has reg metric, remember to use --low_is_better:

  • python run_experiment.py --exp_mode mcts --task househouse_prices --rollout 5 --low_is_better

Custom Experimenter

Code and Configs Explanation

datasets.yaml 提供数据集对应的指标和基础提示词

data.yaml 继承了datasets.yaml以及一些路径信息,需要将datasets_dir指到数据集合集的根目录下

完整的DI提示词参考dataset.py中的generate_task_requirement函数

Evaluation

evaluation.py 提供pred和原始的gt1D iterable以及需要使用的metric返回evaluation score