nemo_rl.data.eval_datasets.math#

Math dataset and its variants.

Module Contents#

Classes#

MathDataset

API#

class nemo_rl.data.eval_datasets.math.MathDataset(
variant: Literal[math_test, math_500_test] = 'math_test',
prompt_file: Optional[str] = None,
system_prompt_file: Optional[str] = None,
)[source]#

Initialization

_rekey(data: dict[str, Any])[source]#