Skip to content

Commit

Permalink
Add beeswarm example to README
Browse files Browse the repository at this point in the history
  • Loading branch information
snehankekre committed Feb 1, 2022
1 parent 3b3ddd8 commit 6ab2975
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -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)
Expand Down

0 comments on commit 6ab2975

Please sign in to comment.