CommencezCommencez gratuitement

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

Voir le cours

Instructions de l’exercice

  • Calculez les shap_values à l'aide d'un TreeExplainer.
  • Utilisez les shap_values obtenus 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
____
Modifier et exécuter le code