Nemo Rl Data | Nvidia

Sous-packages

Sous-modules

interfaces
datasets
multimodal_utils
processors - Contient des processeurs de données pour l’évaluation.
llm_message_utils

Contenu du module

Classes

Nom Description

DataConfig dict() -> nouveau dictionnaire vide dict(mapping) -> nouveau dictionnaire initialisé à partir des paires (clé, valeur) d’un objet de mappage dict(iterable) -> nouveau dictionnaire initialisé comme si via : d = {} for k, v in iterable: d[k] = v dict(**kwargs) -> nouveau dictionnaire initialisé avec les paires nom=valeur dans la liste des arguments de mots-clés. Par exemple : dict(one=1, two=2)

MathDataConfig dict() -> nouveau dictionnaire vide dict(mapping) -> nouveau dictionnaire initialisé à partir des paires (clé, valeur) d’un objet de mappage dict(iterable) -> nouveau dictionnaire initialisé comme si via : d = {} for k, v in iterable: d[k] = v dict(**kwargs) -> nouveau dictionnaire initialisé avec les paires nom=valeur dans la liste des arguments de mots-clés. Par exemple : dict(one=1, two=2)

Nom	Description
`DataConfig`	dict() -> nouveau dictionnaire vide dict(mapping) -> nouveau dictionnaire initialisé à partir des paires (clé, valeur) d’un objet de mappage dict(iterable) -> nouveau dictionnaire initialisé comme si via : d = {} for k, v in iterable: d[k] = v dict(**kwargs) -> nouveau dictionnaire initialisé avec les paires nom=valeur dans la liste des arguments de mots-clés. Par exemple : dict(one=1, two=2)
`MathDataConfig`	dict() -> nouveau dictionnaire vide dict(mapping) -> nouveau dictionnaire initialisé à partir des paires (clé, valeur) d’un objet de mappage dict(iterable) -> nouveau dictionnaire initialisé comme si via : d = {} for k, v in iterable: d[k] = v dict(**kwargs) -> nouveau dictionnaire initialisé avec les paires nom=valeur dans la liste des arguments de mots-clés. Par exemple : dict(one=1, two=2)

API

class nemo_rl.data.DataConfig

Bases : typing.TypedDict

max_input_seq_length: int

Valeur : None

prompt_file: typing.NotRequired[str]

Valeur : None

system_prompt_file: typing.NotRequired[str]

Valeur : None

dataset_name: str

Valeur : None

val_dataset_name: typing.NotRequired[str]

Valeur : None

add_bos: typing.NotRequired[bool]

Valeur : None

add_eos: typing.NotRequired[bool]

Valeur : None

input_key: typing.NotRequired[str]

Valeur : None

output_key: typing.NotRequired[str]

Valeur : None

add_generation_prompt: typing.NotRequired[bool]

Valeur : None

add_system_prompt: typing.NotRequired[bool]

Valeur : None

split: typing.NotRequired[str]

Valeur : None

shuffle: typing.NotRequired[bool]

Valeur : None

seed: typing.NotRequired[int]

Valeur : None

download_dir: typing.NotRequired[str]

Valeur : None

train_data_path: typing.NotRequired[str]

Valeur : None

val_data_paths: typing.NotRequired[dict[str, str]]

Valeur : None

class nemo_rl.data.MathDataConfig

Bases : nemo_rl.data.DataConfig

problem_key: str

Valeur : None

solution_key: str

Valeur : None