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

Feat/change ntree to iteration range #40

Closed GabrielGimenez closed 2 years ago

GabrielGimenez commented 2 years ago

ntree_limits is deprecated in recent versions of xgboost, I've replaced it with iteration range