From 23ce9cccddecc816dce56baa97790d746e3154a9 Mon Sep 17 00:00:00 2001 From: Guillaume VIGNAL Date: Fri, 8 Dec 2023 10:01:08 +0100 Subject: [PATCH] fix bug on TreeExplainer selection --- shapash/backend/shap_backend.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/shapash/backend/shap_backend.py b/shapash/backend/shap_backend.py index ec6867af..4771758c 100644 --- 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)