mirror of
https://github.com/open-thought/reasoning-gym.git
synced 2026-04-19 12:58:07 +00:00
Generate eval config tool (#240)
* feat: Add generate_config.py script to create eval configurations
This commit is contained in:
parent
850c1cf6f4
commit
5b8d1b5175
9 changed files with 858 additions and 338 deletions
|
|
@ -97,6 +97,22 @@ categories:
|
|||
- dataset: "word_sorting"
|
||||
```
|
||||
|
||||
### Generating Configurations
|
||||
|
||||
You can generate a configuration file with all registered datasets using the `generate_config.py` script:
|
||||
|
||||
```bash
|
||||
python generate_config.py --output my_config.yaml --model "anthropic/claude-3.5-sonnet" --provider "Anthropic" --size 50 --seed 42
|
||||
```
|
||||
|
||||
Options:
|
||||
- `--output`: Output YAML file path (default: all_datasets.yaml)
|
||||
- `--model`: Model name (default: openai/gpt-4)
|
||||
- `--provider`: Provider name (default: None)
|
||||
- `--size`: Default dataset size (default: 100)
|
||||
- `--seed`: Default dataset seed (default: 42)
|
||||
- `--include-params`: Include all configuration parameters (default: False)
|
||||
|
||||
### Running Evaluations
|
||||
|
||||
To run evaluations:
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue