1. Apprendre
    2. /
  2. Cours
    3. /
  3. Introduction au Deep Learning avec PyTorch
Connected

Exercice

Calculer la justesse avec torchmetrics

Suivre la justesse pendant l'entraînement aide à repérer l'époque la plus performante.

Dans cet exercice, vous allez utiliser torchmetrics pour calculer la justesse sur un jeu de données de masques avec trois classes. La fonction plot_errors mettra en évidence les échantillons mal classés afin de vous aider à analyser les erreurs du modèle.

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

Instructions 1/2

undefined XP
    1
    2
  • Créez une métrique de justesse "multiclass" avec trois classes.
  • Calculez la justesse pour chaque lot du chargeur de données (dataloader).