Pengarang: Hrishikesh Patel
Awalnya diterbitkan di Towards AI the World’s Leading AI and Technology News and Media Company. Jika Anda sedang membangun produk atau layanan terkait AI, kami mengundang Anda untuk mempertimbangkan untuk menjadi sponsor AI. Di Towards AI, kami membantu menskalakan AI dan startup teknologi. Biarkan kami membantu Anda melepaskan teknologi Anda kepada massa.
Cara Unik Memvisualisasikan Confusion Matrix — Sankey Chart
Pergi Sankey untuk Kurang Kebingungan!
Gambar oleh penulis
Matriks kebingungan dalam pembelajaran mesin dengan mudah merangkum kinerja model. Namun, ketika berkomunikasi dengan pemangku kepentingan non-teknis, matriks kebingungan mungkin tampak tidak intuitif . Jadi apa solusinya — buat diagram Sankey.
Diagram Sankey mewakili matriks kebingungan biner (gambar oleh penulis
Gambar di atas mengilustrasikan diagram Sankey untuk matriks kebingungan biner yang khas. Dalam diagram,
Kotak persegi panjang di sebelah kiri menunjukkan kelas Benar sedangkan kotak di sebelah kanan menunjukkan kelas yang Diprediksi. Warna hijau menyoroti klasifikasi yang benar dan warna merah untuk kesalahan klasifikasi.
Garis besar cerita
Apa diagram Sankey secara singkat? Bagaimana Cara Membuat diagram Sankey dari Matriks Kebingungan? Bonus
Apa diagram Sankey secara singkat?
Diagram Sankey digunakan untuk memvisualisasikan aliran atau koneksi dari sumber ke sink. Mari kita pahami penerapannya dengan contoh sederhana.
Pertimbangkan kami memiliki kumpulan data pendaftaran👨🎓👩🎓 dalam ilmu data atau kursus analisis bisnis di tiga universitas🏫. Di sini universitas dapat diperlakukan sebagai sumber dan mata kuliah sebagai wastafel. Jumlah pendaftaran menunjukkan koneksi dari sumber ke wastafel. Beberapa koneksi ini bisa lebih berat daripada yang lain, misalnya koneksi dari Universitas A ke Ilmu Data lebih berat daripada koneksinya ke analitik bisnis.
Diagram Sankey dibuat dari https://sankeymatic.com/build/
Diagram Sankey untuk matriks konfusi memiliki komponen sebagai berikut:
Sumber: Target Kelas Sejati (Sink): Kelas yang Diprediksi Koneksi/aliran: Jumlah instance
Bagaimana Cara Membuat diagram Sankey dari Matriks Kebingungan?
Kami akan mengikuti pendekatan 3 langkah seperti yang diilustrasikan pada gambar di bawah ini untuk membuat diagram Sankey.
3 langkah pendekatan untuk memplot diagram Sankey dari Confusion Matrix (Gambar oleh penulis)
Langkah-1: Dapatkan Matriks Kebingungan
Pada langkah ini, kita akan membuat matriks konfusi. Ini bisa menjadi output dari fungsi sci-kit belajar confusion_matrix. Untuk mempermudah, kita akan menggunakan matriks konfusi berikut.
https://medium.com/media/4f13b9f93acc3b567d0f94aad0dcfcef/href
Langkah-2: Ubah Matriks Kebingungan menjadi DataFrame
Kami akan membagi langkah ini menjadi beberapa langkah kecil.
2.1 — Buat kerangka data dari matriks kebingungan
https://medium.com/media/b29c7b97f64daa09cd7563647a174c32/href
2.2 — Merestrukturisasi kerangka data
https://medium.com/media/f7abb513ba6b2479d5b3833d859d61ab/href
2.3 — Tambahkan ‘warna’ kolom baru
Sekarang kita akan menambahkan kolom baru ‘warna’ untuk menyoroti kebenaran prediksi. Di sini rgba(211.255.216.0.6) menunjukkan warna hijau, yang akan menyoroti prediksi yang benar. Sedangkan prediksi yang salah akan disorot dengan warna merah yaitu rgba(245.173.168,0.6) .
https://medium.com/media/c028985f3f631ac0eed4510f22a71f00/href
2.4 — Memetakan sumber dan kolom target ke indeks numerik
https://medium.com/media/052a1f13d78abda924ccac0800f47172/href
Mari tambahkan kolom baru agar teks ditampilkan saat kita mengarahkan kursor ke bagan.
2.5 — Tambahkan “keterangan alat” Kolom Baru
https://medium.com/media/d87dabfe18b76cd5384092146a929a31/href
Sekarang kita siap untuk memplot grafik.
Langkah-3: Buat Bagan Sankey
Fungsi plotting go.Sankey mengambil dua argumen utama — node & link. Node adalah kelas yaitu True Class 1, Predicted Class 2, dll. dan link adalah koneksi/aliran antara True dan Predicted Class.
https://medium.com/media/7561a4b800899f97becbbca44a58ba6b/href
Bonus
Terima kasih telah menyelesaikan semua langkah😀 Saya tahu ini adalah proses yang membosankan untuk melewati langkah-langkahnya, jadi mengapa tidak membuat fungsi? Yah, saya telah membuat fungsi praktis untuk memplot bagan Sankey untuk matriks kebingungan apa pun (biner & multi-kelas).
Jangan ragu untuk memeriksa notebook ini di GitHub untuk mempelajari lebih lanjut tentang fungsinya.
Sebelum kamu pergi!
Saya harap Anda menikmati ceritanya dan merasa bermanfaat. Ikuti saya di Medium jika Anda ingin lebih banyak cerita seperti ini dan berlangganan saya untuk mendapatkan cerita baru saya langsung ke kotak masuk Anda.
Cerita saya yang lain, Anda mungkin menikmati …
Cara Unik Memvisualisasikan Matriks Kebingungan — Sankey Chart awalnya diterbitkan di Towards AI on Medium, di mana orang-orang melanjutkan percakapan dengan menyoroti dan menanggapi cerita ini.
Diterbitkan melalui Menuju AI
