CommencerCommencer gratuitement

Calcul de la précision à l’aide des métriques torch (torchmetrics)

Le suivi de la précision pendant l’apprentissage permet d’identifier l’époque la plus performante.

Dans cet exercice, vous utiliserez torchmetrics pour calculer la précision sur un ensemble de données de masque de visage à trois classes. La fonction plot_errors met en évidence les échantillons mal classés, ce qui vous aide à analyser les erreurs de modèle.

Le paquet torchmetrics est déjà importé. Les outputs du modèle sont constitués de probabilités softmax et les labels sont constitués de vecteurs encodés one-hot.

Cet exercice fait partie du cours

Introduction au deep learning avec PyTorch

Afficher le cours

Exercice interactif pratique

Essayez cet exercice en complétant cet exemple de code.

# Create accuracy metric
metric = torchmetrics.____(____, ____)
for features, labels in dataloader:
    outputs = model(features)
  
    # Calculate accuracy over the batch
    metric.____(____, ____)
Modifier et exécuter le code