Merge pull request #36 from Aarhus-Psychiatry-Research/mb/add_ci_to_t…

…imedelta feat: add ci to timedelta plots
Aarhus-Psychiatry-Research · May 11, 2023 · c8aeac3 · c8aeac3
2 parents 4873aec + 232dc93
commit c8aeac3
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/psycop/model_evaluation/binary/time/timedelta_plots.py b/src/psycop/model_evaluation/binary/time/timedelta_plots.py
@@ -69,8 +69,11 @@ def plot_roc_auc_by_time_from_first_visit(
         bin_unit=bin_unit,
         bin_continuous_input=bin_continuous_input,
         drop_na_events=False,
+        confidence_interval=0.95,
     )
 
+    ci = df["ci"].tolist()
+
     bin_unit2str = {
         "h": "Hours",
         "D": "Days",
@@ -90,6 +93,7 @@ def plot_roc_auc_by_time_from_first_visit(
         plot_type=["line", "scatter"],
         bar_count_values=df["n_in_bin"],
         bar_count_y_axis_title="Number of visits",
+        confidence_interval=ci,
         save_path=save_path,
     )