nemo_rl.data.eval_datasets.local_math_dataset

Local math dataset.

Module Contents

Classes

LocalMathDataset

API

class nemo_rl.data.eval_datasets.local_math_dataset.LocalMathDataset(

data_paths: str | list[str],

problem_key: str,

solution_key: str,

name: str,

split: Optional[str] = None,

file_format: Literal[csv, json] = 'csv',

prompt_file: Optional[str] = None,

system_prompt_file: Optional[str] = None,

)

Initialization

_rekey(data: dict[str, Any])