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
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.____(____, ____)