`nemo_rl.data.datasets.raw_dataset`#

Module Contents#

Classes#

API#

class nemo_rl.data.datasets.raw_dataset.RawDataset#

data_config: nemo_rl.data.ResponseDatasetConfig#: None

dataset: datasets.Dataset#: None

val_dataset: datasets.Dataset | None#: None

processor: nemo_rl.data.interfaces.TaskDataProcessFnCallable#: None

task_spec: nemo_rl.data.interfaces.TaskDataSpec#: None

split_train_validation(test_size: float, seed: int)#

set_processor()#

set_task_spec(data_config: nemo_rl.data.ResponseDatasetConfig)#