`nemo_rl.data.datasets.response_datasets.squad`#

Module Contents#

Classes#

Simple wrapper around the squad dataset.

API#

class nemo_rl.data.datasets.response_datasets.squad.SquadDataset(split: str = 'train', **kwargs)#

Bases: nemo_rl.data.datasets.raw_dataset.RawDataset

Simple wrapper around the squad dataset.

Parameters:: split – Split name for the dataset, default is “train”

Initialization

format_data(data: dict[str, Any]) → dict[str, Any]#