nemo_rl.data.datasets.preference_datasets

Submodules

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

Package Contents

Functions

load_preference_dataset

Loads preference dataset.

Data

__all__

API

nemo_rl.data.datasets.preference_datasets.load_preference_dataset(data_config)

Loads preference dataset.

nemo_rl.data.datasets.preference_datasets.__all__

[‘BinaryPreferenceDataset’, ‘HelpSteer3Dataset’, ‘PreferenceDataset’, ‘Tulu3PreferenceDataset’]