Menghitung akurasi menggunakan torchmetrics
Melacak akurasi selama pelatihan membantu mengidentifikasi epoch dengan kinerja terbaik.
Dalam latihan ini, Anda akan menggunakan torchmetrics untuk menghitung akurasi pada himpunan data facemask dengan tiga kelas. Fungsi plot_errors akan menyoroti sampel yang salah klasifikasi, membantu Anda menganalisis kesalahan model.
Paket torchmetrics sudah diimpor. outputs model berupa probabilitas softmax, dan labels berupa vektor one-hot.
Latihan ini adalah bagian dari kursus
Pengantar Deep Learning dengan PyTorch
Latihan interaktif praktis
Cobalah latihan ini dengan menyelesaikan kode contoh berikut.
# Create accuracy metric
metric = torchmetrics.____(____, ____)
for features, labels in dataloader:
outputs = model(features)
# Calculate accuracy over the batch
metric.____(____, ____)