【问题标题】:Store compiled model in Keras?在 Keras 中存储编译模型?
【发布时间】:2016-08-17 20:54:14
【问题描述】:

我想在生产中使用 Keras 模型,但不想在每次需要运行它时都重新编译模型。我应该只是腌制它等吗?

【问题讨论】:

  • Theano 或 TensorFlow 后端?

标签: python neural-network keras


【解决方案1】:

如果您只想预测(而不是在生产中训练),则无需编译模型。

来源:https://github.com/fchollet/keras/issues/2621

【讨论】:

    【解决方案2】:

    很遗憾,Keras 目前不支持 pickle。您可以:

    • 使用内置的“keras.models.load_model”和“keras.models.save_model”将模型和训练后的权重存储到 hdf5 文件中。
    • 使用不完整的技巧使 Keras 模型“可腌制”。如果您有一个引用模型的对象,这很有用。我做了一次,并在我的博客上发了一篇文章 - http://zachmoshe.com/2017/04/03/pickling-keras-models.html

    【讨论】:

      猜你喜欢
      • 2021-08-30
      • 1970-01-01
      • 1970-01-01
      • 2019-12-23
      • 1970-01-01
      • 2022-01-06
      • 1970-01-01
      • 1970-01-01
      • 1970-01-01
      相关资源
      最近更新 更多