Diagrammes en violon

Utilisez la fonction plot_violin() pour créer un diagramme en violon SHAP. Ceci peut être utilisé pour visualiser la distribution et l’intervalle des valeurs SHAP pour chaque fonctionnalité.

Exemple de diagramme en violon montrant les distributions d'influence des fonctionnalités pour toutes les fonctionnalités

Arguments obligatoires

Argument

Description

shap_df

Tableau 2D contenant des valeurs SHAP pour plusieurs fonctionnalités

feature_df

Tableau 2D contenant les valeurs de fonctionnalités correspondantes

Arguments facultatifs

Argument

Description

figsize

Un tuple de (largeur, hauteur) qui contrôle la taille du diagramme. Utilise une taille par défaut de (1 400, 100) si non spécifié.

La fonction renvoie un graphique qui représente un diagramme en violon pour chaque fonctionnalité. Les diagrammes en violon sont organisés selon la moyenne absolue de la valeur SHAP de chaque fonctionnalité, les fonctionnalités ayant la plus forte influence sur les prédictions du modèle étant placées en tête.