xgboost-survival-embeddings icon indicating copy to clipboard operation
xgboost-survival-embeddings copied to clipboard

Feat/add pre trained model xgbse 1st step

Open davivieirab opened this issue 2 years ago • 0 comments

  • 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

davivieirab avatar Aug 08 '22 23:08 davivieirab