Visualiser l'importance des caractéristiques
Dans cet exercice, vous allez déterminer quelles caractéristiques ont été les plus prédictives selon le régleur de forêts aléatoires rf que vous avez entraîné dans un exercice précédent.
Pour ce faire, vous allez tracer un diagramme à bandes horizontal représentant l'importance des caractéristiques telle qu'évaluée par rf. Heureusement, cela se fait facilement grâce aux fonctionnalités de traçage de pandas.
Nous avons créé un objet pandas.Series appelé importances qui contient les noms des caractéristiques comme index et leurs importances comme valeurs. De plus, matplotlib.pyplot est disponible sous le nom plt et pandas sous pd.
Cette activité fait partie du cours
Machine Learning avec des modèles à base d'arbres en Python
Instructions de l’exercice
Appelez la méthode
.sort_values()surimportanceset assignez le résultat àimportances_sorted.Appelez la méthode
.plot()surimportances_sortedet définissez les arguments :kindà'barh'colorà'lightgreen'
Exercice interactif pratique
Essayez cet exercice en complétant ce code d’exemple.
# Create a pd.Series of features importances
importances = pd.Series(data=rf.feature_importances_,
index= X_train.columns)
# Sort importances
importances_sorted = ____
# Draw a horizontal barplot of importances_sorted
____.____(____='____', ____='____')
plt.title('Features Importances')
plt.show()