Merge remote-tracking branch 'origin/expo' into optimize-mle-bench-prompt

This commit is contained in:
Yizhou Chi 2024-10-18 16:02:57 +08:00
commit 345d79979d

View file

@ -157,46 +157,15 @@ #### Setup
git clone https://github.com/WecoAI/aideml.git
```
Modify `aideml/aide/utils/config.yaml`:
Modify `aideml/aide/utils/config.yaml` - change `k_fold_validation`, `code model`, and `feedback model` as follows:
```yaml
# path to the task data directory
data_dir: null
# either provide a path to a plaintext file describing the task
desc_file: null
# or provide the task goal (and optionally evaluation information) as arguments
goal: null
eval: null
log_dir: logs
workspace_dir: workspaces
# whether to unzip any archives in the data directory
preprocess_data: True
# whether to copy the data to the workspace directory (otherwise it will be symlinked)
# copying is recommended to prevent the agent from accidentally modifying the original data
copy_data: True
exp_name: null # a random experiment name will be generated if not provided
# settings for code execution
exec:
timeout: 3600
agent_file_name: runfile.py
format_tb_ipython: False
# agent hyperparams
agent:
# how many improvement iterations to run
steps: 10
# whether to instruct the agent to use CV (set to 1 to disable)
k_fold_validation: 1
# whether to instruct the agent to generate a prediction function
expose_prediction: False
# whether to provide the agent with a preview of the data
data_preview: True
# LLM settings for coding
code:
model: deepseek-coder