如何将我的 keras 模型保存到谷歌驱动器或计算机中？

Posted 2023-02-16

【中文标题】如何将我的 keras 模型保存到谷歌驱动器或计算机中？

【英文标题】：How do I save my keras model into google drive or the computer?

【发布时间】：2021-12-11 09:58:11

【问题描述】：

没有代码， 我正在制作一个像样的 ML 项目，我想将我的 keras 模型导入谷歌驱动器或将其保存在我的计算机中，以便我可以立即使用“model.predict()”函数。 你能帮帮我吗？

【参考方案1】：

你可以使用 model.save() 和 model.load()

更多细节和例子here

如果您使用的是 google colab，您可以将您的 google drive 安装在 notebook（左侧菜单）中，然后直接从驱动器中保存和加载。

【讨论】：

我已经知道了。

那些方法不能解决你的问题？你的问题能更具体一点吗？

好吧，训练一个模型需要很长时间，所以我想保存一个模型，以便我可以毫不拖延地进行预测。

但这正是 model.save 所做的，当您使用 model.save 保存模型时，权重已包含在内，您无需再次对其进行训练。只需使用 model = keras.models.load_model('path') 然后 model.predict(input)。只要确保使用 .h5 格式