Merge branch 'improve_mcts' into 'expo'

add fixed ss and special instructions

See merge request agents/exp_optimizer!15
This commit is contained in:
林义章 2024-09-14 13:44:28 +00:00
commit 3cba031c2c
12 changed files with 130 additions and 56 deletions

View file

@ -223,16 +223,10 @@ ### Base DI
For setup, check 5.
- `python run_experiment.py --exp_mode base --task titanic --num_experiments 10`
- Ask DI to use AutoGluon: `--special_instruction ag`
- Ask DI to use the stacking ensemble method: `--special_instruction stacking`
### DI RandomSearch
For setup, check 5.
- 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`
## 5. DI MCTS
@ -259,6 +253,20 @@ #### Run
- `python run_experiment.py --exp_mode mcts --task househouse_prices --rollout 10 --low_is_better`
In addition to the generated insights, include the fixed insights saved in `expo/insights/fixed_insights.json`
- `--use_fixed_insights`
#### Ablation Study
**DI RandomSearch**
- 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`