set check_additivity=False for CausalForestDML

py-why · Apr 30, 2021 · ea9e01d · ea9e01d
1 parent 9630421
commit ea9e01d
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/econml/_shap.py b/econml/_shap.py
@@ -333,8 +333,7 @@ def _shap_explain_multitask_model_cate(cme_model, multitask_model_cate, X, d_t,
                                      output_names=output_names_,
                                      input_names=input_names_,
                                      background_samples=background_samples)
-
-        shap_out = explainer(F)
+        shap_out = explainer(F, check_additivity=False)
         if dt > 1:
             for i in range(dt):
                 base_values = shap_out.base_values[..., i]