CommencezCommencez gratuitement

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

Voir le cours

Instructions de l’exercice

  • Parcourez les importances des caractéristiques produites par rfr.
  • Affichez le nom de chaque colonne de X_train et 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[____], ____))
Modifier et exécuter le code