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
321 stars 53 forks source link

Using a Pretrained XGBoost Model #51

Open BardiaKh opened 2 years ago

BardiaKh commented 2 years ago

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.

davivieirab commented 2 years ago

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.

BardiaKh commented 2 years ago

Great!