mirror of
https://github.com/open-thought/reasoning-gym.git
synced 2026-04-19 12:58:07 +00:00
updated read me
This commit is contained in:
parent
bea806fe3c
commit
046c46c0bb
14 changed files with 193 additions and 438 deletions
|
|
@ -1,25 +0,0 @@
|
|||
from dataclasses import dataclass
|
||||
from typing import Union
|
||||
|
||||
import yaml
|
||||
|
||||
from reasoning_gym.utils import SYSTEM_PROMPTS
|
||||
|
||||
|
||||
@dataclass
|
||||
class EvalConfig:
|
||||
category: str
|
||||
datasets: Union[str, list[str]]
|
||||
eval_dir: str
|
||||
dataset_size: int
|
||||
dataset_seed: int
|
||||
model: str = "deepseek/deepseek-r1"
|
||||
provider: str = "Nebius"
|
||||
developer_role: str = "system"
|
||||
developer_prompt: str = SYSTEM_PROMPTS["DeepSeekZero"]
|
||||
|
||||
@classmethod
|
||||
def from_yaml(cls, yaml_path: str):
|
||||
with open(yaml_path, "r") as f:
|
||||
config = yaml.safe_load(f)
|
||||
return cls(**config)
|
||||
Loading…
Add table
Add a link
Reference in a new issue