Save the trained models about lightautoml HOT 5 CLOSED

Hi @niyuzheno1 ,

Are you looking for a specific way to save models?

from lightautoml.

Hi @niyuzheno1,
To save the model you can use joblib.dump(automl, ‘model.pkl’).
To load the model you can use automl=joblib.load(‘model.pkl’)

Both these commands need import joblib at the begging of the script.

from lightautoml.

I want to save my model whenever a submodel is trained. More specifically, I want to have the functionality that suppose we have models such as lgb, lgb_tuned, linear_l2, cb. Whenever one of the models get trained, we will save the current version of the model.

from lightautoml.

@niyuzheno1 I wonder if this article might further your cause? AnalyticsIndiaMag - Custom Pipeline

from lightautoml.

@niyuzheno1 we have investigate different ideas about saving the models separately, but it's just not possible because LightAutoML is not just the models - it's the full pipeline from the raw data to the predictions and even further. That's why we save the whole pipeline at the end and use it for the prediction.

from lightautoml.