CommencezCommencez gratuitement

Calculer l'importance des caractéristiques avec les arbres de décision

Vous avez construit un classificateur par arbre de décision pour repérer les patients à risque de maladie cardiaque à partir du jeu de données sur les maladies cardiaques. Vous devez maintenant expliquer le modèle en analysant l'importance des caractéristiques afin de déterminer les facteurs clés de prédiction de la maladie cardiaque, ce qui permettra des interventions de santé plus ciblées.

matplotlib.pyplot a été importé sous le nom plt. X_train et y_train sont déjà chargés pour vous.

Cette activité fait partie du cours

Explainable AI en Python

Voir le cours

Instructions de l’exercice

  • Extrayez l'importance des caractéristiques à partir de model.
  • Tracez les feature_importances pour les feature_names fournis.

Exercice interactif pratique

Essayez cet exercice en complétant ce code d’exemple.

model = DecisionTreeClassifier(random_state=42)
model.fit(X_train, y_train)

# Derive feature importances
feature_importances = ____
feature_names = X_train.columns

# Plot the feature importances
____
plt.show()
Modifier et exécuter le code