search

sb-ai-lab / LightAutoML

Fast and customizable framework for automatic ML model creation (AutoML)
https://developers.sber.ru/portal/products/lightautoml
Apache License 2.0
1.08k stars 47 forks source link

Inference image after save weight. #129

Closed DaniilAnd closed 5 months ago

DaniilAnd commented 9 months ago

Hi! How to load a model and make a predict ? Interested inference image.

dev-rinchin commented 9 months ago

Hi, you can dump and load using pickle. Check example in demo0.py in lines 162-170: https://github.com/sb-ai-lab/LightAutoML/blob/fddb8a5c96f0274e1a586af4062936969b2593fa/examples/demo0.py#L162