Meningkatkan plot
Agar plot lebih mudah dibaca, kita perlu mencapai dua tujuan:
- Mengurutkan batang secara menaik.
- Menambahkan label pada plot yang sesuai dengan nama fitur.
Untuk melakukannya, kita akan memanfaatkan pengindeksan NumPy. Metode .argsort() mengurutkan sebuah array dan mengembalikan indeksnya. Kita akan menggunakan indeks ini untuk mencapai kedua tujuan tersebut.
Latihan ini adalah bagian dari kursus
Analitik Pemasaran: Memprediksi Churn Pelanggan di Python
Petunjuk latihan
- Hitung indeks terurut dari
importancesdengan menggunakannp.argsort()padaimportances. - Buat label terurut dengan mengakses kolom
Xdan melakukan pengindeksan menggunakansorted_index. - Buat plot dengan mengindeks
importancesmenggunakansorted_indexdan tentukan argumen kata kuncitick_label=labels.
Latihan interaktif praktis
Cobalah latihan ini dengan menyelesaikan kode contoh berikut.
# Sort importances
sorted_index = ____(____)
# Create labels
labels = X.columns[____]
# Clear current plot
plt.clf()
# Create plot
plt.barh(range(X.shape[1]), importances[____], tick_label=____)
plt.show()