在Keras存储编译模型?

时间:2016-08-17 20:54:14

标签: python neural-network keras

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

2 个答案:

答案 0 :(得分:2)

如果你想做的只是预测(而不是生产中的训练),就没有必要编译模型。

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

答案 1 :(得分:1)

目前,Keras不支持pickle。你可以:

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