-
Notifications
You must be signed in to change notification settings - Fork 12
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
API change the shape of the incidence curves #61
API change the shape of the incidence curves #61
Conversation
MAINT activate doctest check with pytest by default
The goal is to be consistent with the conventions of scikit-learn where all |
Fix a pandas warning in load_seer
FIX do not pytest collect outside of the hazardous module folder
CI trigger build doc on all targeted branches
…ck_predict_survival_boost
hazardous/_survival_boost.py
Outdated
The first column holds the survival probability to any event and others the | ||
incicence probabilities for each event. | ||
predicted_curves : ndarray of shape (n_samples, n_events + 1, n_times) | ||
The estimated probabilities at different time horizons. The column |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
"column" is not meaningful any more for a 3D array.
The values for event with index 0 are the estimated probabilities of staying event-free at the requested time horizons for each observation described by the matching row of X. ...
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Other than that LGTM.
Change some API such that the shape of returned arrays are:
predict_proba
is an array of shape (n_samples, n_events)predict_cumulative_incidence
is an array of shape (n_samples, n_events, n_times)predict_survival_function
is an array of shape (n_samples, n_times)