Closed bamblebam closed 3 years ago
Hi @bamblebam,
The best trained model is already inside automl object, which you can use to do prediction for the test data.
If you want to save the the model, you can use joblib
library: joblib.dump(automl, 'trained_lightautoml_model.pkl')
To load the model the code automl = joblib.load('trained_lightautoml_model.pkl')
can be used.
Hope this helps.
Alex
Thank You for replying.
So I just trained a model on some tabular data using TabularUtilizedAutoML and wanted to save the model. But I couldn't find anything related to saving the best trained model.
Just wanted to know if there is a functionality to do so ?