更改型号:< name>由keras在model.summary()输出中自动给出 [英] change the Model: <name> given automatically by keras in model.summary() output

问题描述

调用命令时:

print(model.summary())

我得到以下输出:

如何重命名Keras自动生成的突出显示的字段?

How can I rename the highlighted field, which is generated automatically by 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()

如果您有经过预先训练的模型，则可以轻松完成

if u have a pre-trained model you can simply do

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

这篇关于更改型号:< name>由keras在model.summary()输出中自动给出的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！