Fix no convergence XGBTimeSeriesSurvival tests

vanderschaarlab · Mar 15, 2023 · 1e3d6b4 · 1e3d6b4
1 parent cf6ea56
commit 1e3d6b4
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/synthcity/metrics/eval_performance.py b/src/synthcity/metrics/eval_performance.py
@@ -675,7 +675,7 @@ def evaluate(
                     "n_jobs": 2,
                     "verbosity": 0,
                     "depth": 3,
-                    "strategy": "weibull",  # "weibull", "debiased_bce"
+                    "strategy": "debiased_bce",  # "weibull", "debiased_bce"
                     "random_state": self._random_state,
                 },
                 X_gt,
@@ -705,7 +705,7 @@ def evaluate(
                     "n_jobs": 2,
                     "verbosity": 0,
                     "depth": 3,
-                    "strategy": "weibull",  # "weibull", "debiased_bce"
+                    "strategy": "debiased_bce",  # "weibull", "debiased_bce"
                     "random_state": self._random_state,
                 },
                 X_gt,
@@ -1033,7 +1033,7 @@ def evaluate(
                 n_jobs=2,
                 verbosity=0,
                 depth=3,
-                strategy="weibull",  # "weibull", "debiased_bce"
+                strategy="debiased_bce",  # "weibull", "debiased_bce"
                 random_state=self._random_state,
             )