MetaGPT/expo/README.md

58 lines
1.3 KiB
Markdown
Raw Normal View History

2024-08-30 20:35:17 +08:00
# Expo
## Setup
In the root directory, `pip install -e .`
`cd expo`
`pip install -r requirements.txt`
2024-08-30 20:35:17 +08:00
## Instruction
- 下载数据集https://deepwisdom.feishu.cn/drive/folder/RVyofv9cvlvtxKdddt2cyn3BnTc?from=from_copylink
- 修改`data.yaml``datasets_dir`为数据集合集根目录存储位置
2024-08-30 20:35:17 +08:00
## 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`
2024-09-02 20:00:46 +08:00
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`
2024-08-30 20:35:17 +08:00
## Custom Experimenter
2024-08-30 20:35:17 +08:00
2024-09-02 17:01:25 +08:00
## Code and Configs Explanation
2024-08-30 20:35:17 +08:00
2024-09-02 17:01:25 +08:00
`datasets.yaml` 提供数据集对应的指标和基础提示词
`data.yaml` 继承了`datasets.yaml`以及一些路径信息,需要将`datasets_dir`指到数据集合集的根目录下
完整的DI提示词参考`dataset.py`中的`generate_task_requirement`函数
## Evaluation
`evaluation.py` 提供pred和原始的gt1D iterable以及需要使用的metric返回evaluation score
2024-08-30 20:35:17 +08:00