Evaluasi akurasi model menggunakan Torchmetrics
Menilai seberapa baik kinerja model Anda sangat penting—terutama saat menyiapkannya untuk deployment! Mari integrasikan perhitungan akurasi dengan mulus menggunakan Torchmetrics langsung di dalam validation_step(). Jangan lupa untuk melakukan pencatatan (log) hasilnya, sehingga Anda dapat memantau kemajuan model dengan mudah.
Latihan ini adalah bagian dari kursus
Model AI yang Dapat Diskalakan dengan PyTorch Lightning
Petunjuk latihan
- Impor
Accuracydaritorchmetrics. - Inisialisasi metrik akurasi di dalam
__init__(). - Hitung akurasi di dalam
validation_step()dan catat (log) sebagai'val_acc'.
Latihan interaktif praktis
Cobalah latihan ini dengan menyelesaikan kode contoh berikut.
# 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)