Keras模型没有保存

时间:2017-01-20 10:46:08

标签: keras

当我通过model.save(filepath)在Keras中保存我的模型时,文件会被保存并且没有错误,但是当我打开文件时这就是我得到的:

enter image description here

我在网上搜索过,似乎没有人遇到这个问题。我的控制台也没有细节。有什么想法吗?

3 个答案:

答案 0 :(得分:3)

要加载模型,请使用此

from keras.models import load_model
model = load_model(filepath)

您可以找到更多文档here

答案 1 :(得分:3)

保存模型

# serialize model to JSON
model_json = model.to_json()
with open("model.json", "w") as json_file:             
     json_file.write(model_json) 

# serialize weights to HDF5
model.save_weights("model.h5")
print("Saved model to disk")

加载模型

from keras.models import load_model
model.load_weights("model.h5")

答案 2 :(得分:2)

以下是YouTube视频,其中详细说明了您想要做的事情:Save and load a Keras model

Keras提供三种不同的保存方法。这些在上面的视频链接(带有示例)以及下面进行了描述。

首先,您收到错误的原因是因为文本编辑器无法读取HDF5文件。

您使用了正确的model.save(filepath)功能。此保存功能可以保存:

  • 模型的架构,允许重新创建模型。
  • 模特的重量。
  • 培训配置(损失,优化程序)。
  • 优化器的状态,允许您从中断的位置恢复训练。

要加载此已保存的模型,请使用以下命令:

    from keras.models import load_model
    new_model = load_model(filepath)

如果您只使用model.to_json(),则只会保存模型的体系结构。此外,如果您使用model.save_weights(),则只能保存模型的权重。使用这两种替代保存技术,您不会保存训练配置(丢失,优化器),也不会保存优化器的状态。