From ea9e01def265c80bd8eab1cb9313bb1bb745c379 Mon Sep 17 00:00:00 2001 From: Maggie Hei Date: Fri, 30 Apr 2021 15:48:53 -0400 Subject: [PATCH] set check_additivity=False for CausalForestDML --- econml/_shap.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/econml/_shap.py b/econml/_shap.py index da2ed2cb3..0442b4595 100644 --- 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]