Analyser les effets des variables avec des beeswarm plots
Dans votre rôle de data scientist à l’université, vous vous concentrez maintenant sur une analyse plus fine des effets de chaque variable sur les résultats d’admission. Identifier les facteurs clés qui influencent les décisions d’admission était essentiel, mais aller plus loin permet de comprendre comment les variations de ces facteurs affectent précisément les prédictions. Ces informations supplémentaires aideront à répondre à des questions telles que l’impact d’une modification des scores de test ou du CGPA sur la probabilité d’admission, afin d’éclairer les décisions et les recommandations de politique.
La bibliothèque shap et les données d’entraînement (X_train, y_train) ont été préchargées pour vous.
Cet exercice fait partie du cours
IA explicable en Python
Instructions
- Obtenez les
shap_valuesà l’aide d’unTreeExplainer. - Utilisez les
shap_valuesobtenues pour tracer le beeswarm plot 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
____