SHAP Summary Plot and Mean Values displaying together-CodePudding

Used the following Python code for a SHAP Summary Plot:

explainer = shap.TreeExplainer(model2)
shap_values = explainer.shap_values(X_sampled)
shap.summary_plot (shap_values, X_sampled, max_display=X_sampled.shape[1])

and got a plot which is something like this:

while in R, the plot looks like:

How can I modify my Python script to include mean (|SHAP value|) corresponding to each feature in the same plot (just like the R output)?

CodePudding user response：

SHAP plots are a bit tricky to customize unless you're willing to tinker with the source code, but the following will do:

import xgboost
import shap

X, y = shap.datasets.adult()
model = xgboost.XGBClassifier().fit(X, y)
explainer = shap.TreeExplainer(model, X)
shap_values = explainer(X)
feature_names = [
    a   ": "   str(b) for a,b in zip(X.columns, np.abs(shap_values.values).mean(0).round(2))
]

shap.summary_plot(shap_values, X, max_display=X.shape[1], 
                  feature_names=feature_names)