From 9d2bbb9b1575ca99a2c05fce053ca989c2c75393 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Francesc=20Mart=C3=AD=20Escofet?= <154450563+FrancescMartiEscofetQC@users.noreply.github.com> Date: Mon, 24 Jun 2024 13:36:17 +0200 Subject: [PATCH] Rename --- metalearners/tlearner.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/metalearners/tlearner.py b/metalearners/tlearner.py index 875a986..11dacaa 100644 --- a/metalearners/tlearner.py +++ b/metalearners/tlearner.py @@ -118,7 +118,6 @@ def evaluate( ) -> dict[str, float]: safe_scoring = self._scoring(scoring) - return _evaluate_model_kind( cfes=self._nuisance_models[VARIANT_OUTCOME_MODEL], Xs=[X[w == tv] for tv in range(self.n_variants)], @@ -127,5 +126,5 @@ def evaluate( model_kind=VARIANT_OUTCOME_MODEL, is_oos=is_oos, oos_method=oos_method, - is_treatment=False, + is_treatment_model=False, )