Menganalisis metrik per kelas
Meskipun metrik teragregasi berguna sebagai indikator kinerja model, sering kali lebih informatif untuk melihat metrik per kelas. Ini dapat mengungkap kelas-kelas di mana model berkinerja kurang baik.
Dalam latihan ini, Anda akan menjalankan kembali loop evaluasi untuk memperoleh precision dari pengklasifikasi awan kita, tetapi kali ini per kelas. Lalu, Anda akan memetakan skor tersebut ke nama kelas untuk menafsirkannya. Seperti biasa, Precision sudah diimpor untuk Anda. Semoga berhasil!
Latihan ini adalah bagian dari kursus
Deep Learning Lanjutan dengan PyTorch
Petunjuk latihan
- Definisikan metrik precision yang sesuai untuk hasil per kelas.
- Hitung precision per kelas dengan melengkapi dict comprehension, mengiterasi
.items()dari atribut.class_to_idxpadadataset_test.
Latihan interaktif praktis
Cobalah latihan ini dengan menyelesaikan kode contoh berikut.
# Define precision metric
metric_precision = Precision(
____, ____, ____
)
net.eval()
with torch.no_grad():
for images, labels in dataloader_test:
outputs = net(images)
_, preds = torch.max(outputs, 1)
metric_precision(preds, labels)
precision = metric_precision.compute()
# Get precision per class
precision_per_class = {
k: ____[____].____
for k, v
in ____
}
print(precision_per_class)