MulaiMulai sekarang secara gratis

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

Lihat Kursus

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_idx pada dataset_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)
Edit dan Jalankan Kode