nemo_rl.data.hf_datasets.deepscaler#

Module Contents#

Classes#

Functions#

format_math

prepare_deepscaler_dataset

Load and split the DeepScaler dataset into train and test sets.

API#

nemo_rl.data.hf_datasets.deepscaler.format_math(data)[source]#
nemo_rl.data.hf_datasets.deepscaler.prepare_deepscaler_dataset(seed=42)[source]#

Load and split the DeepScaler dataset into train and test sets.

class nemo_rl.data.hf_datasets.deepscaler.DeepScalerDataset(seed: int = 42)[source]#

Initialization

Initialize the DeepScaler dataset with train/test split.

Parameters:

seed – Random seed for reproducible splitting