diff --git a/README.md b/README.md index c5a009d..56d33fb 100644 --- a/README.md +++ b/README.md @@ -52,6 +52,7 @@ explainer = shap.Explainer(model, X) shap_values = explainer(X) st_shap(shap.plots.waterfall(shap_values[0]), height=300) +st_shap(shap.plots.beeswarm(shap_values), height=300) explainer = shap.TreeExplainer(model) shap_values = explainer.shap_values(X)