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

fix:predictions use best tree when early stopping #27

Closed GabrielGimenez closed 3 years ago

GabrielGimenez commented 3 years ago

Status

READY

Todo list

Fix predictions to use the best trees when using early stopping