且构网

分享程序员开发的那些事...
且构网 - 分享程序员编程开发的那些事

有没有一种方法可以重命名Keras模型的指标和损失?

更新时间:2023-12-01 22:15:10

每个_loss_accuracy之前的名称均来自输出层的名称.如果要修改此名称,则应重命名输出层.

The names before each _loss and _accuracy came from the names of output layers. If you want to modify this names you should rename the output layers.

请考虑以下模型.

input_ =  keras.layers.Input(shape=(8,))
x =  keras.layers.Dense(16)(input_)
output1 = keras.layers.Dense(32, name="output1")(x)
output2 = keras.layers.Dense(32, name="output2")(x)
model = keras.models.Model(inputs=input_, outputs=[output1, output2])
model.compile(loss=["mse", "mae"], optimizer="adam", metrics={"output1":"accuracy","output2":"accuracy"})

现在model.metrics_names将为您提供以下列表

Now model.metrics_names will give you the following list

['loss', 'output1_loss', 'output2_loss', 'output1_acc', 'output2_acc']