CommencerCommencer gratuitement

Calculer l’importance des variables avec les arbres de décision

Vous avez entraîné un classificateur par arbre de décision pour identifier les patient·e·s à risque de maladie cardiaque à partir du jeu de données correspondant. Vous devez maintenant expliquer le modèle en analysant l’importance des variables afin de déterminer les facteurs clés de prédiction des maladies cardiaques, pour permettre des interventions de santé plus ciblées.

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

Cet exercice fait partie du cours

IA explicable en Python

Afficher le cours

Instructions

  • Extrayez les importances des variables à partir du model.
  • Tracez les feature_importances pour les feature_names fournis.

Exercice interactif pratique

Essayez cet exercice en complétant cet exemple de code.

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