Importances des caractéristiques
Même si certains attributs de bonbons, comme le chocolat, sont très populaires, cela ne veut pas dire qu'ils seront importants pour la prédiction du modèle. Après l'ajustement d'un modèle de forêt aléatoire, vous pouvez consulter l'attribut du modèle, .feature_importances_, pour voir quelles variables ont eu le plus grand impact. Vous pouvez vérifier l'importance de chaque variable dans le modèle en itérant sur le tableau des importances au moyen de enumerate().
Si vous ne connaissez pas la fonction enumerate() de Python, elle permet de parcourir une liste tout en créant un compteur automatique.
Cette activité fait partie du cours
Validation de modèles en Python
Instructions de l’exercice
- Parcourez les importances des caractéristiques produites par
rfr. - Affichez le nom de chaque colonne de
X_trainet la cote d'importance correspondante.
Exercice interactif pratique
Essayez cet exercice en complétant ce code d’exemple.
# Fit the model using X and y
rfr.fit(X_train, y_train)
# Print how important each column is to the model
for i, item in enumerate(rfr.____):
# Use i and item to print out the feature importance of each column
print("{0:s}: {1:.2f}".format(X_train.columns[____], ____))