1. Učit se
  2. /
  3. Kurzy
  4. /
  5. Introduction to Deep Learning with PyTorch

Connected

cvičení

Výpočet přesnosti pomocí torchmetrics

Sledování přesnosti během trénování ti pomůže identifikovat nejlépe fungující epochu.

V tomto cvičení použiješ torchmetrics k výpočtu přesnosti na datasetu roušek se třemi třídami. Funkce plot_errors zvýrazní chybně klasifikované vzorky a pomůže ti analyzovat chyby modelu.

Balíček torchmetrics je již importován. Výstupy modelu outputs jsou pravděpodobnosti ze softmaxu a labels jsou vektory zakódované metodou one-hot.

Pokyny 1/2

undefined XP
    1
    2
  • Vytvoř metriku přesnosti typu "multiclass" se třemi třídami.
  • Vypočítej přesnost pro každou dávku z dataloaderem.