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 merupakan bagian dari kursus
Deep Learning Lanjutan dengan PyTorch
Instruksi 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 langsung praktik
Cobalah latihan ini dengan melengkapi kode contoh ini.
# 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)