Set explicit value for n_events

soda-inria · Jul 3, 2024 · 6cb17cd · 6cb17cd
1 parent d262cfe
commit 6cb17cd
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/examples/plot_marginal_cumulative_incidence_estimation.py b/examples/plot_marginal_cumulative_incidence_estimation.py
@@ -54,6 +54,7 @@
 
 X_uncensored, y_uncensored = make_synthetic_competing_weibull(
     n_samples=n_samples,
+    n_events=n_events,
     censoring_relative_scale=0,
     return_X_y=True,
     shape_ranges=shape_ranges,
@@ -213,6 +214,7 @@ def plot_cumulative_incidence_functions(y, survival_boost=None, aj=None):
 # of censoring.
 X_censored, y_censored = make_synthetic_competing_weibull(
     n_samples=n_samples,
+    n_events=n_events,
     censoring_relative_scale=1.5,
     return_X_y=True,
     shape_ranges=shape_ranges,