nemo_rl.data.datasets.preference_datasets

Submodules

• nemo_rl.data.datasets.preference_datasets.preference_dataset
• nemo_rl.data.datasets.preference_datasets.binary_preference_dataset
• nemo_rl.data.datasets.preference_datasets.tulu3
• nemo_rl.data.datasets.preference_datasets.helpsteer3

Package Contents

Functions

load_preference_dataset

Loads preference dataset.

Data

DATASET_REGISTRY
__all__

API

nemo_rl.data.datasets.preference_datasets.DATASET_REGISTRY

None

nemo_rl.data.datasets.preference_datasets.load_preference_dataset(

data_config: nemo_rl.data.PreferenceDatasetConfig,

)

Loads preference dataset.

nemo_rl.data.datasets.preference_datasets.__all__

[‘BinaryPreferenceDataset’, ‘HelpSteer3Dataset’, ‘PreferenceDataset’, ‘Tulu3PreferenceDataset’, ‘loa…