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

Using a Pretrained XGBoost Model

Open BardiaKh opened this issue 2 years ago • 2 comments

Describe the feature and the current state.

In the XGBSEDebiasedBCE model, there is no way to overwrite the model with an already-trained model. This is useful as XGBSE does not expose some low-level APIs of the XGBoost model, for example, sample/feature weighting. This feature would be useful to add the data and an already trained model and then run the LR on top of that model.

Will this change a current behavior? How?

No. It will only add a method to the XGBSEDebiasedBCE model to replace the 'self.bst' and run the multitask LR task.

BardiaKh avatar Mar 03 '22 22:03 BardiaKh

Thanks for the report, @BardiaKh . We are currently working on a way to replace the 1st step xgboost model from both XGBSEDebiasedBCE and XGBSEStackedWeibull modules. Both will be part of our next minor release.

davivieirab avatar Aug 02 '22 17:08 davivieirab

Great!

BardiaKh avatar Aug 02 '22 20:08 BardiaKh