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/add pre trained model xgbse 1st step #60

Open davivieirab opened 1 year ago

davivieirab commented 1 year ago

Add tests:

Add docs: