Analyse des effets des caractéristiques à l'aide de graphiques en forme d'essaim d'abeilles
Poursuivant votre rôle de data scientist à l'université, vous vous concentrez désormais sur une analyse plus détaillée des effets des caractéristiques individuelles sur les résultats des admissions. Si l'identification des facteurs clés influençant les décisions d'admission était cruciale, un examen plus approfondi nous permet de comprendre comment les variations de ces facteurs affectent spécifiquement les prédictions. Ces informations supplémentaires permettront de répondre à des questions telles que la manière dont les modifications des résultats des tests ou de la moyenne générale influencent la probabilité d'admission, ce qui permettra de disposer d'une image plus claire pour prendre des décisions éclairées et formuler des recommandations politiques.
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 dérivé
shap_values
pour tracer le graphique de l'essaim d'abeilles et l'analyser.
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 beeswarm plot
____