代码之家  ›  专栏  ›  技术社区  ›  Hagbard

保存和检索模型名称

  •  0
  • Hagbard  · 技术社区  · 7 年前

    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 回复  |  直到 7 年前
        1
  •  1
  •   today    7 年前

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