issues
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/add pre trained model xgbse 1st step
#60
Open
davivieirab
opened
1 year ago
davivieirab
commented
1 year ago
Add the possibility of using a pre-trained XGB estimator for the 1st step model in both XGBSEDebiasedBCE and XGBSEStackedWeibull
Add tests:
function to assert a valid xgb estimator is used as pre-trained model
model with and without the pre-trained xgb is fitted properly for both modules
Add docs:
example on how to use a pre-trained xgb model for both modules
Add tests:
Add docs: