ÎncepețiÎncepe gratuit

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.

Acest exercițiu face parte din cursul

Introducere în Deep Learning cu PyTorch

Vezi cursul

Exercițiu interactiv practic

Încearcă acest exercițiu completând acest cod de exemplu.

# Create accuracy metric
metric = torchmetrics.____(____, ____)
for features, labels in dataloader:
    outputs = model(features)
  
    # Calculate accuracy over the batch
    metric.____(____, ____)
Editează și rulează codul