Pass fit_cate_intercept correctly

py-why · Aug 11, 2020 · d11b6d8 · d11b6d8
1 parent 4c1bb3a
commit d11b6d8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/econml/ortho_iv.py b/econml/ortho_iv.py
@@ -1414,7 +1414,7 @@ def __init__(self, model_Y_X, model_T_XZ,
         super().__init__(model_Y_X, model_T_XZ,
                          flexible_model_effect=flexible_model_effect,
                          featurizer=featurizer,
-                         fit_cate_intercept=True,
+                         fit_cate_intercept=fit_cate_intercept,
                          final_model_effect=StatsModelsLinearRegression(fit_intercept=False),
                          cov_clip=cov_clip, n_splits=n_splits, opt_reweighted=opt_reweighted,
                          categories=categories)