更改模型：在model.summary()输出中由Keras自动生成的模型名称

当我调用以下命令时：

print(model.summary())

我得到了以下输出：

如何重命名Keras自动生成的高亮显示的字段？

提前感谢您的帮助。

回答：

有一个参数 ‘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' # 如果由于TF版本问题导致错误，请尝试使用m.namem.summary()

学技术