我的模型有一个输出。但我想将损失分成3个不同的组件,以便我可以跟踪每个组件的进度。有没有办法用keras做到这一点?
也许我可以定义与指标相同的损失组件,但是有更优雅的方式吗?
您可以将损失定义为Keras图层,然后您可以手动添加所有损失和指标(如果需要)。
You can see a complete tutorial about this topic here
TL; DR:
None
作为损失这是在代码中手动添加损失的方式:
loss_layer_names = {"my_loss", ...}
# Adding losses
for name in loss_layer_names:
layer = model.get_layer(name)
loss = (tf.reduce_mean(layer.output, keepdims=True))
model.add_loss(loss)
# Adding metrics
for name in loss_layer_names:
layer = model.get_layer(name)
loss = (tf.reduce_mean(layer.output, keepdims=True))
model.metrics_names.append(name)
model.metrics_tensors.append(loss)
model.compile(optimizer="adam", loss=[None] * len(model.outputs))
model
是Keras模型