Merge remote-tracking branch 'origin/main' into mb/add_ci_to_timedelta

Aarhus-Psychiatry-Research · May 11, 2023 · 232dc93 · 232dc93
2 parents ce8c63f + 2c01d58
commit 232dc93
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/src/psycop/model_evaluation/binary/utils.py b/src/psycop/model_evaluation/binary/utils.py
@@ -11,7 +11,7 @@ def calc_performance(
     df: pd.DataFrame,
     metric: Callable,
     confidence_interval: Optional[float] = None,
-    n_bootstraps: int = 1000,
+    n_bootstraps: int = 100,
     **kwargs: Any,
 ) -> pd.Series:
     """Calculates performance metrics of a df with 'y' and 'input_to_fn' columns.
@@ -59,7 +59,11 @@ def metric_wrapper(
         ) -> float:
             # bootstrap function requires the metric function to
             # be able to take additional arguments (notably the length of the array)
-            return metric(true, pred)
+            try:
+                return metric(true, pred)
+            except ValueError as e:
+                print(repr(e))
+                return np.nan
 
         boot = bootstrap(
             (df["y"], df["y_hat"]),