nemo_rl.data.datasets.response_datasets.daily_omni#
Module Contents#
Classes#
Simple wrapper around the Daily-Omni dataset. |
API#
- class nemo_rl.data.datasets.response_datasets.daily_omni.DailyOmniDataset(
- split: str = 'train',
- split_validation_size: float = 0,
- seed: int = 42,
- **kwargs,
Bases:
nemo_rl.data.datasets.raw_dataset.RawDatasetSimple wrapper around the Daily-Omni dataset.
- Parameters:
split – Split name for the dataset, default is “train”
Initialization
- task_name#
‘daily-omni’
- classmethod get_prompt(data: dict[str, Any]) str#
- format_data(data: dict[str, Any]) dict[str, Any]#