nemo_rl.data.datasets.eval_datasets

Submodules

• nemo_rl.data.datasets.eval_datasets.mmlu_pro
• nemo_rl.data.datasets.eval_datasets.gpqa
• nemo_rl.data.datasets.eval_datasets.local_math_dataset
• nemo_rl.data.datasets.eval_datasets.aime
• nemo_rl.data.datasets.eval_datasets.mmlu
• nemo_rl.data.datasets.eval_datasets.math

Package Contents

Functions

load_eval_dataset

Loads evaluation dataset.

Data

__all__

API

nemo_rl.data.datasets.eval_datasets.load_eval_dataset(data_config)

Loads evaluation dataset.

nemo_rl.data.datasets.eval_datasets.__all__

[‘AIMEDataset’, ‘GPQADataset’, ‘LocalMathDataset’, ‘MathDataset’, ‘MMLUDataset’, ‘MMLUProDataset’]