nemo_rl.data.datasets.response_datasets.dapo_math

Module Contents

Classes

DAPOMath17KDataset	Simple wrapper around the DAPO Math 17K dataset with train split.
DAPOMathAIME2024Dataset

API

class nemo_rl.data.datasets.response_datasets.dapo_math.DAPOMath17KDataset(**kwargs)

Bases: nemo_rl.data.datasets.raw_dataset.RawDataset

Simple wrapper around the DAPO Math 17K dataset with train split.

Initialization

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

class nemo_rl.data.datasets.response_datasets.dapo_math.DAPOMathAIME2024Dataset(**kwargs)

Bases: nemo_rl.data.datasets.response_datasets.dapo_math.DAPOMath17KDataset