Passer à la navigation

nemo_rl.data.hf_datasets.deepscaler

Afficher en Markdown

Contenu du module

Classes

NomDescription
DeepScalerDatasetAucun

Fonctions

NomDescription
format_mathAucun
prepare_deepscaler_datasetCharger et diviser le jeu de données DeepScaler en ensembles d’entraînement et de test.

API

nemo_rl.data.hf_datasets.deepscaler.format_math(data: dict[str, str | float | int]) -> dict[str, list[typing.Any] | str]
nemo_rl.data.hf_datasets.deepscaler.prepare_deepscaler_dataset(seed: int = 42) -> dict[str, datasets.Dataset | None]

Charger et diviser le jeu de données DeepScaler en ensembles d’entraînement et de test.

class nemo_rl.data.hf_datasets.deepscaler.DeepScalerDataset(seed: int = 42)