我正在使用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'
。我在这里想念什么?
答案 0 :(得分:1)
显然,通过将Keras(从2.2.0版升级到2.2.4版),此问题得以解决。