2024-08-30 20:35:17 +08:00
|
|
|
|
# Expo
|
|
|
|
|
|
|
|
|
|
|
|
|
2024-09-02 20:23:45 +08:00
|
|
|
|
## 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
|
2024-09-02 20:23:45 +08:00
|
|
|
|
- 修改`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
|
|
|
|
|
2024-09-02 20:23:45 +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和原始的gt(1D iterable)以及需要使用的metric,返回evaluation score
|
2024-08-30 20:35:17 +08:00
|
|
|
|
|