预训练机器学习

时间:2019-03-15 20:05:16

标签: python machine-learning keras

我使用Keras库为图像分类建立了CNN模型。但是培训需要很多小时。训练模型后,如何在不进行训练的情况下使用它?我的意思是说,在训练模型后,我想多次使用它。 因为我将在android studio中使用我的模型。 感谢您的帮助

谢谢

1 个答案:

答案 0 :(得分:2)

在训练过程之后,您可以使用以下方法轻松保存模型:

model.save('my_model.h5')

您以后可以使用以下方式加载该模型:

model = load_model('my_model.h5')

有关更多详细信息,请参阅文档:https://keras.io/getting-started/faq/#how-can-i-save-a-keras-model