首页 文章

Keras重命名模型和图层

提问于
浏览
0

1)我尝试使用TF后端重命名模型和Keras中的图层,因为我在一个脚本中使用多个模型 . 类Model似乎具有属性model.name,但在更改它时我得到“AttributeError:无法设置属性” . 这里有什么问题?

2)此外,我正在使用顺序API,我想给图层命名,这似乎是功能API的可能,但我找不到顺序API的解决方案 . anonye知道如何为顺序API做到这一点吗?

更新到2):命名图层工作,虽然它似乎没有记录 . 只需添加参数名称,例如model.add(Dense(...,...,name = "hiddenLayer1") . 注意,具有相同名称的图层共享权重!

3 回答

  • -1

    关于型号名称的第一个问题在我的机器上无法重现 . 我可以像这样设置它 . 很多时候这些错误都是由软件版本引起的 .

    model=Sequential()
    model.add(Dense(2,input_shape=(....)))
    model.name="NAME"
    

    至于命名图层,您可以在顺序模型中执行此操作

    model=Sequential()
    model.add(Dense(2,input_shape=(...),name="NAME"))
    
  • 3

    来自user239457的Answer仅适用于Standard keras .

    如果你想使用Tensorflow Keras,你可以这样做:

    from tensorflow.keras.models import Sequential
    from tensorflow.keras.layers import Dense
    
    model = Sequential(name='Name')
    model.add(Dense(2,input_shape=(5, 1)))
    
  • 0

    对于1),我认为你可以 Build 另一个具有正确名称和与现有模型相同结构的模型 . 然后将存在模型的图层的权重设置为新模型的图层 .

相关问题