Update src/med_bench/estimation/base.py

judithabk6 · Dec 19, 2024 · dce928d · dce928d
1 parent b22d50f
commit dce928d
Showing 1 changed file with 0 additions and 1 deletion.
diff --git 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