Skip to content

Instantly share code, notes, and snippets.

@betterdatascience

betterdatascience/explainers.py

Created December 14, 2020 11:27
Show Gist options
  • Save betterdatascience/65db591e2c840940d6ea5480feccbe05 to your computer and use it in GitHub Desktop.
Save betterdatascience/65db591e2c840940d6ea5480feccbe05 to your computer and use it in GitHub Desktop.
Download ZIP

Revisions

  1. betterdatascience created this gist Dec 14, 2020.
    7 changes: 7 additions & 0 deletions explainers.py
    View file
    Original file line number Diff line number Diff line change
    @@ -0,0 +1,7 @@
    import shap
    shap.initjs()

    shap_explainer = shap.TreeExplainer(model)
    shap_values = shap_explainer.shap_values(X)

    shap.force_plot(shap_explainer.expected_value, shap_values[1, :], test_1)