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 merupakan bagian dari kursus
Pengantar Deep Learning dengan PyTorch
Latihan interaktif langsung praktik
Cobalah latihan ini dengan melengkapi kode contoh ini.
# Create accuracy metric
metric = torchmetrics.____(____, ____)
for features, labels in dataloader:
outputs = model(features)
# Calculate accuracy over the batch
metric.____(____, ____)