remove mlebench

This commit is contained in:
Cyzus Chi 2024-10-31 00:05:44 +08:00
parent 261a39d547
commit 776ee4455c

View file

@ -165,34 +165,4 @@ ### 5. Custom Baselines
To run additional baselines:
- Each baseline must produce `dev_predictions.csv` and `test_predictions.csv` with a `target` column.
- Use the `evaluate_score` function for evaluation.
---
## MLE-Bench
**Note:** MLE-Bench requires Python 3.11 or higher.
#### Setup
Clone the repository and install:
```bash
git clone https://github.com/openai/mle-bench.git
cd mle-bench
pip install -e .
```
Prepare the data:
```bash
mlebench prepare -c <competition-id> --data-dir <dataset-dir-save-path>
```
#### Run the MLE-Bench Experiment
Run the following command to execute the experiment:
```bash
python run_experiment.py --exp_mode mcts --custom_dataset_dir <dataset-dir-save-path/prepared/public> --rollouts 10 --from_scratch --role_timeout 3600
```
- Use the `evaluate_score` function for evaluation.