Začněte nyníZačněte zdarma

Vyhodnocení přesnosti modelu pomocí Torchmetrics

Vyhodnocení výkonu modelu je klíčové – zvlášť před nasazením do produkce! Pojďme plynule zapojit výpočet přesnosti pomocí Torchmetrics přímo do metody validation_step(). Nezapomeň výsledky zalogovat, abys mohl/a snadno sledovat, jak se tvůj model zlepšuje.

Toto cvičení je součástí kurzu

Škálovatelné modely AI s PyTorch Lightning

Zobrazit kurz

Pokyny k cvičení

  • Importuj Accuracy z torchmetrics.
  • Vytvoř instanci metriky přesnosti uvnitř __init__().
  • Vypočítej přesnost v metodě validation_step() a zaloguj ji pod názvem 'val_acc'.

Interaktivní cvičení na vyzkoušení si v praxi

Vyzkoušejte si toto cvičení dokončením tohoto ukázkového kódu.

# Import relevant metric
from torchmetrics import ____
import lightning.pytorch as pl

class ClassifierModel(pl.LightningModule):
    def __init__(self):
        super().__init__()
        # Instantiate accuracy metric
        self.accuracy = ____()
    def validation_step(self, batch, batch_idx):
        x, y = batch
        preds = self(x)
        # Calculate accuracy and log it as val_acc
        acc = self.____(preds, y)
        self.log(____, acc)
Upravit a spustit kód