1. Learn
  2. /
  3. Cursuri
  4. /
  5. Introducere în Deep Learning cu PyTorch

Connected

exercițiu

Calcularea acurateței cu torchmetrics

Urmărirea acurateței în timpul antrenamentului te ajută să identifici epoca cu cele mai bune rezultate.

În acest exercițiu, vei folosi torchmetrics pentru a calcula acuratețea pe un set de date cu măști faciale cu trei clase. Funcția plot_errors va evidenția eșantioanele clasificate greșit, ajutându-te să analizezi erorile modelului.

Pachetul torchmetrics este deja importat. outputs ale modelului sunt probabilități softmax, iar labels sunt vectori codificați one-hot.

Instrucțiuni 1/2

undefined XP
    1
    2
  • Creează o metrică de acuratețe "multiclass" cu trei clase.
  • Calculează acuratețea pentru fiecare lot din dataloader.