Analyser les effets des caractéristiques avec des diagrammes en essaim
Dans votre rôle de spécialiste des données à l'université, vous vous concentrez maintenant sur une analyse plus détaillée des effets de chaque caractéristique sur les résultats d'admission. Repérer 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 comme la manière dont les changements aux résultats de test ou au CGPA influencent la probabilité d'admission, offrant un portrait plus clair pour prendre des décisions éclairées et formuler des recommandations de politiques.
La bibliothèque shap et les données d'entraînement (X_train, y_train) ont été préchargées pour vous.
Cette activité fait partie du cours
Explainable AI en Python
Instructions de l’exercice
- Calculez les
shap_valuesà l'aide d'unTreeExplainer. - Utilisez les
shap_valuesobtenus pour tracer le diagramme en essaim et l'analyser.
Exercice interactif pratique
Essayez cet exercice en complétant ce code d’exemple.
model = RandomForestRegressor(random_state=42)
model.fit(X_train, y_train)
# Derive shap values
explainer = ____
shap_values = ____
# Plot the beeswarm plot
____