sberbank-ai-lab/LightAutoML

save model

lk137095576 opened this issue · 2 comments

how to save and load model config or weight?

Hi @lk137095576,
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.

Alex

Hi @lk137095576,
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.

Alex

thx