CommencezCommencez gratuitement

Calculer la perte d'entropie croisée

La perte d'entropie croisée est une méthode largement utilisée pour mesurer la perte en classification. Dans cet exercice, vous calculerez l'entropie croisée dans PyTorch à l'aide de :

  • y : l'étiquette véritable (ground truth).
  • scores : un vecteur de prédictions avant softmax.

Les fonctions de perte aident les réseaux neuronaux à apprendre en mesurant les erreurs de prédiction. Créez un vecteur à codage one-hot pour y, définissez la fonction de perte d'entropie croisée et calculez la perte à partir de scores et de l'étiquette encodée. Le résultat sera un seul nombre à virgule flottante représentant la perte de l'exemple.

torch, CrossEntropyLoss et torch.nn.functional sous le nom F ont déjà été importés pour vous.

Cette activité fait partie du cours

Introduction au Deep Learning avec PyTorch

Voir le cours

Exercice interactif pratique

Essayez cet exercice en complétant ce code d’exemple.

import torch
import torch.nn.functional as F
from torch.nn import CrossEntropyLoss

y = [2]
scores = torch.tensor([[0.1, 6.0, -2.0, 3.2]])

# Create a one-hot encoded vector of the label y
one_hot_label = F.____(torch.____(____), num_classes=____)
Modifier et exécuter le code