CommencerCommencer gratuitement

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

Afficher le cours

Instructions

  • Extraire les importances des caractéristiques à partir du site model.
  • Tracez le graphique feature_importances pour le site feature_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()
Modifier et exécuter le code