De plot verbeteren
Om de plot leesbaarder te maken, willen we twee doelen bereiken:
- Sorteer de staafjes in oplopende volgorde.
- Voeg labels toe aan de plot die overeenkomen met de featurenamen.
Hiervoor maken we gebruik van NumPy-indexering. De methode .argsort() sorteert een array en geeft de indexen terug. We gebruiken deze indexen om beide doelen te bereiken.
Deze oefening maakt deel uit van de cursus
Marketinganalyse: klantverloop voorspellen in Python
Oefeninstructies
- Bereken de gesorteerde indexen van
importancesdoornp.argsort()opimportancestoe te passen. - Maak de gesorteerde labels door de kolommen van
Xte benaderen en te indexeren metsorted_index. - Maak de plot door
importanceste indexeren metsorted_indexen het keywordargumenttick_label=labelsop te geven.
Praktische interactieve oefening
Probeer deze oefening eens door deze voorbeeldcode in te vullen.
# 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()