更改Keras中model.summary()输出中自动给定的模型名称<name>

Question

调用命令时：

print(model.summary())

我得到以下输出：

我该如何重命名Keras自动生成的高亮字段？谢谢您的帮助。

- NikSp

Answer 1

这里有一个参数 'name'。

在函数格式中。

inp = Input((10,))
out = Dense(1)(inp)

m = Model(inp, out, name='model_XXX')
m.summary()

以顺序方式排列

m = Sequential([Dense(1, input_dim=10)], name='model_XXX')
m.summary()

如果您拥有一个预先训练好的模型，您可以简单地执行以下操作：

m.fit(...)
m._name = 'model_XXX' # try with m.name if it raise error due to TF version
m.summary()

- Marco Cerliani

---

谢谢您的回答，keras.Model(name="model_XXX") 工作得很好 :) - NikSp

Answer 2

如果您想重命名已经构建好的模型，可以按照以下步骤进行：

model.name = 'yourname'

- Dr. Snoopy

---

已经尝试过这个方法，但是model.name命令会改变Sequential()对象的名称，而我想要改变Model()对象的名称。对于造成的不便，非常抱歉。刚刚我发现Sequential()和Model()对象具有不同的模型名称。无论如何，非常感谢您的回答 :) - NikSp

@NikSp 我不确定你在说什么，我用一个模型实例尝试了一下，它运行良好。 - Dr. Snoopy

嗯，好的，可能我做错了什么。我会仔细检查并告诉你。 - NikSp