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
Instructions
- Extrayez les importances des variables à partir du
model. - Tracez les
feature_importancespour lesfeature_namesfournis.
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()