nemo_rl.data.hf_datasets.prompt_response_dataset

Module Contents

Classes

PromptResponseDataset

API

class nemo_rl.data.hf_datasets.prompt_response_dataset.PromptResponseDataset(

train_ds_path: str,

val_ds_path: str,

input_key: str = 'input',

output_key: str = 'output',

)

Initialization

add_messages_key(example)