nemo_rl.data.datasets.response_datasets.helpsteer3

Module Contents

Classes

HelpSteer3Dataset

Simple wrapper around the HelpSteer3 dataset with preference subset.

API

class nemo_rl.data.datasets.response_datasets.helpsteer3.HelpSteer3Dataset(split: str = 'train', **kwargs)

Bases: nemo_rl.data.datasets.raw_dataset.RawDataset

Simple wrapper around the HelpSteer3 dataset with preference subset.

Parameters:

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

Initialization

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