CommencerCommencer gratuitement

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

Afficher le cours

Instructions

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