CommencerCommencer gratuitement

Graphiques d'importance des caractéristiques pour l'analyse des admissions

En tant que membre de l'équipe de science des données d'une université, votre tâche consiste à évaluer les facteurs qui déterminent réellement les décisions d'admission et ceux qui sont moins importants. Si le comité d'admission sait que la moyenne générale joue un rôle clé, il souhaite le confirmer et découvrir d'autres facteurs importants susceptibles d'influencer les résultats. À l'aide d'un RandomForestRegressor model, vous visualiserez l'importance des caractéristiques afin d'identifier clairement les aspects des profils des candidats qui comptent le plus et ceux qui ont le moins d'impact sur le processus de décision.

La bibliothèque shap et les données d'apprentissage (X_train, y_train) ont été préchargées pour vous.

Cet exercice fait partie du cours

L'IA explicable en Python

Afficher le cours

Instructions

  • Dérivez le site shap_values à l'aide d'un TreeExplainer.
  • Utilisez le site shap_values dérivé pour tracer les importances d'éléments à l'aide d'un diagramme à barres et analysez-le.

Exercice interactif pratique

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

model = RandomForestRegressor(random_state=42)
model.fit(X_train, y_train)

# Derive shap values
explainer = ____
shap_values = ____

# Plot the feature importance plot
____
Modifier et exécuter le code