search

loft-br / xgboost-survival-embeddings

Improving XGBoost survival analysis with embeddings and debiased estimators
https://loft-br.github.io/xgboost-survival-embeddings/
Apache License 2.0
313 stars 51 forks source link

Need to get individual survival distributions for censored subjects #30

Closed ericgqc closed 3 years ago

ericgqc commented 3 years ago

Background context

We want to predict the individual survival distribution for censored subjects.

Description of the changes proposed in the pull request

Is there a method of implementing this in xgbse?

GabrielGimenez commented 3 years ago

@ericgqc You can just train the model as usual and use the predict method on the censored subjects. Also, issues are the appropriate place for these questions, pull requests are used to make code contributions to the project.