From dce928d65a7fe4c92a54de17a4079ff7f00e7634 Mon Sep 17 00:00:00 2001 From: bthirion Date: Thu, 19 Dec 2024 15:01:00 +0100 Subject: [PATCH] Update src/med_bench/estimation/base.py --- src/med_bench/estimation/base.py | 1 - 1 file changed, 1 deletion(-) diff --git a/src/med_bench/estimation/base.py b/src/med_bench/estimation/base.py index 84f0a80..e62fe22 100644 --- a/src/med_bench/estimation/base.py +++ b/src/med_bench/estimation/base.py @@ -302,7 +302,6 @@ def _estimate_treatment_propensity_xm(self, m, x): """ xm = np.hstack((x, m)) - # predict P(T=1|X), P(T=1|X, M) p_xm = self._classifier_t_xm.predict_proba(xm)[:, 1] return p_xm