Question: proper way to save/load XGBSEStackedWeibull model

[yerachoi]

Hi. First of all, thank you for developing and sharing XGBSE, which has been of great help for me. I'd like to know how to properly save/load all the key components of a trained XGBSEStackedWeibull model. I have tried out several saving methods but have failed thus far to retrieve the exact same results from the loaded model.

[GabrielGimenez]

We usually use joblib dump and load to save and load models. It's important to fix xgboost, sklearn and lifelines versions when loading, to avoid some difference in the underlying libraries.

so it should be something like

from joblib import dump, load
from xgbse import XGBSEStackedWeibull

...

model = XGBSEStackedWeibull()
model.fit(x_train, y_train)
dump(model, model_path)

...

model = load(model_path)
predictions = model.predict(x_valid)

I'll add a save and load example to the documentation