Fixing istake in survival rates computation

MaastrichtU-CDS · Aug 23, 2023 · fdfe23f · fdfe23f
1 parent 4c4e24e
commit fdfe23f
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/v6_healthai_dashboard_py/survival.py b/v6_healthai_dashboard_py/survival.py
@@ -36,7 +36,9 @@ def survival_rate(df: pd.DataFrame, cutoff: int, delta: int) -> list:
     all_dead = len(df[df['vital_status'] == 'dead'])
     survival_rates = []
     for time in times:
-        dead = len(df[df['survival_days'] <= time])
+        dead = len(
+            df[(df['survival_days'] <= time) & (df['vital_status'] == 'dead')]
+        )
         alive = (all_dead - dead) + all_alive
         survival_rates.append(alive / len(df))