Vylepšení grafu
Aby byl graf přehlednější, potřebujeme splnit dva cíle:
- Seřadit sloupce vzestupně.
- Přidat do grafu popisky odpovídající názvům příznaků.
Využijeme k tomu indexování v NumPy. Metoda .argsort() seřadí pole a vrátí indexy prvků. Pomocí těchto indexů dosáhneme obou cílů.
Toto cvičení je součástí kurzu
Marketing Analytics: Predicting Customer Churn in Python
Pokyny k cvičení
- Vypočítej seřazené indexy pole
importancespomocínp.argsort()aplikovaného naimportances. - Vytvoř seřazené popisky tak, že přistoupíš ke sloupcům
Xa indexuješ je pomocísorted_index. - Vytvoř graf indexováním pole
importancespomocísorted_indexa zadáním klíčového argumentutick_label=labels.
Interaktivní cvičení na vyzkoušení si v praxi
Vyzkoušejte si toto cvičení dokončením tohoto ukázkového kódu.
# 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()