Passer à la navigation
DéveloppementAPI ReferenceNemo RlDataHf Datasets

nemo_rl.data.hf_datasets.geometry3k

Afficher en Markdown

Contenu du module

Classes

NomDescription
Geometry3KDatasetAucun

Fonctions

NomDescription
format_geometry3k_datasetFormater le jeu de données Geometry3K en un journal de messages de type API OpenAI.
prepare_geometry3k_datasetAucun

API

nemo_rl.data.hf_datasets.geometry3k.format_geometry3k_dataset(
    example: dict[str, typing.Any], return_pil: bool = False
) -> dict[str, typing.Any]

Formater le jeu de données Geometry3K en un journal de messages de type API OpenAI.

nemo_rl.data.hf_datasets.geometry3k.prepare_geometry3k_dataset(
    split: str = 'train', task_name: str = 'geometry3k'
)
class nemo_rl.data.hf_datasets.geometry3k.Geometry3KDataset(split: str = 'train', prompt_file: typing.Optional[str] = None)