保存和检索型号名称

时间:2018-10-17 11:36:14

标签: python json keras save

我正在使用Keras训练模型,这就是我指定其名称的方式:

model = Sequential()
model.name = 'NameOfMyModel'
[...]

然后我将其保存到磁盘,如下所示:

model_json = model.to_json()
with open(pathToModel+".json","w") as json_file:
    json_file.write(model_json)

model.save_weights(pathToModel+'.h5')

稍后,我使用以下代码再次加载它:

json_file = open(path+'.json', 'r')
loaded_model_json = json_file.read()
json_file.close()

loaded_model = model_from_json(loaded_model_json)
loaded_model.load_weights(path+".h5")

我想通过以下操作再次从加载的模型中检索模型名称:

modelName = loaded_model.name

但是,返回的不是字符串'NameOfMyModel',而是返回了字符串'sequential_1'。我在这里想念什么?

1 个答案:

答案 0 :(得分:1)

显然,通过将Keras(从2.2.0版升级到2.2.4版),此问题得以解决。