Calcul de l'importance des caractéristiques à l'aide d'arbres de décision
Vous avez construit un arbre de décision pour identifier les patients présentant un risque de maladie cardiaque à l'aide de l'ensemble 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 des maladies cardiaques, ce qui permettra de mieux cibler les interventions en matière de soins de santé.
matplotlib.pyplot
a été importé en tant que plt
. X_train
et y_train
sont préchargés pour vous.
Cet exercice fait partie du cours
L'IA explicable en Python
Instructions
- Extraire les importances des caractéristiques à partir du site
model
. - Tracez le graphique
feature_importances
pour le sitefeature_names
donné.
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()