nemo_rl.environments.metrics

Contenu du module

Fonctions

API

nemo_rl.environments.metrics.calculate_pass_rate_per_prompt(
    prompts: torch.Tensor, is_correct: torch.Tensor
) -> float

Fonction pour calculer la fraction de prompts ayant au moins une réponse correcte (récompense > 0).

prompts: tensor (b, s) Tenseur des prompts utilisés par le modèle. Peut être sur n’importe quel appareil is_correct: tensor (b,) étiquette booléenne. Peut être sur n’importe quel appareil

Retourne : pass rate : float