Merge pull request #515 from guillaume-vignal/master

fix bug on TreeExplainer selection
MAIF · Dec 8, 2023 · 201033c · 201033c
2 parents 8309882 + 23ce9cc
commit 201033c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/shapash/backend/shap_backend.py b/shapash/backend/shap_backend.py
@@ -26,7 +26,7 @@ def __init__(self, model, preprocessing=None, masker=None, explainer_args=None,
         else:
             if shap.explainers.Linear.supports_model_with_masker(model, self.masker):
                 self.explainer = shap.Explainer(model=model, masker=self.masker)
-            elif shap.explainers.Tree.supports_model_with_masker(model, self.masker):
+            elif shap.explainers.Tree.supports_model_with_masker(model, None):
                 self.explainer = shap.Explainer(model=model)
             elif shap.explainers.Additive.supports_model_with_masker(model, self.masker):
                 self.explainer = shap.Explainer(model=model, masker=self.masker)