FIX accept both float and array-like for time_horizon in predict_proba

soda-inria / hazardous

Competing Risks and Survival Analysis

https://soda-inria.github.io/hazardous/

MIT License

45 stars 11 forks source link

FIX accept both float and array-like for time_horizon in predict_proba #54

Closed glemaitre closed 2 months ago

glemaitre commented 2 months ago

There is a bug in SurvivalBoost.predict_proba. The time_horizon parameter does not currently accept array-like while it should.

glemaitre commented 2 months ago

Could you please fix the shape information of the predict_cumulative_incidence docstring? It lacks the time dimension. I believe it should be (n_samples, n_events + 1, n_time_horizons)

I would do that in a subsequent PR.