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
Instructions
- Dérivez le site
shap_values
à l'aide d'unTreeExplainer
. - 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
____