search

sberbank-ai-lab / LightAutoML

LAMA - automatic model creation framework
Apache License 2.0
887 stars 92 forks source link

save model #55

Closed lk137095576 closed 3 years ago

lk137095576 commented 3 years ago

how to save and load model config or weight?

alexmryzhkov commented 3 years ago

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

lk137095576 commented 3 years ago

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