epeters3 / AutoTensor

Automatically train and tune a neural network on your data. Hyperparameter optimization for free.
1 stars 0 forks source link

Persist best model #1

Open epeters3 opened 5 years ago

epeters3 commented 5 years ago

Once the package finds the best model it can, that model should be persisted for the user to be able to consume in the wild. Include an output statement about the persisted model's path. When used from other python code, the method can return the best model, or a path to it.

epeters3 commented 5 years ago

See here for an example: https://machinelearningmastery.com/save-load-keras-deep-learning-models/. Look at the first code example.