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
Instructions de l’exercice
- Extrayez l'importance des caractéristiques à partir de
model. - Tracez les
feature_importancespour lesfeature_namesfournis.
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()