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
321 stars 53 forks source link

fix: include last tree in iteration range #43

Closed GabrielGimenez closed 2 years ago

GabrielGimenez commented 2 years ago

following xgboost docs, fix iteration range https://xgboost.readthedocs.io/en/latest/python/python_intro.html#prediction