无论是自定义还是从keras导入,我都有多个损失和指标。有没有一种方法可以指定可以将哪些模型输出输入到哪个度量,而不是打印或计算所有模型输出?
答案 0 :(得分:1)
是的,您可以将损耗/度量标准作为字典传递给图层名称到损耗/度量标准。
来自documentation的引用:
损失:...如果模型有多个输出,则可以使用其他输出 通过传递字典或损失列表来使每个输出损失。的 模型将最小化的损失值将是 所有个人损失。
和
指标:...为不同的指标指定不同的指标 多输出模型的输出,您还可以传递字典, 例如metrics = {'output_a':'accuracy'}。
示例:
model.compile(
optimizer='rmsprop',
loss={'output_1': 'loss_1', 'output_2': 'loss_2'},
loss_weights={'output_1': 1., 'output_2': 0.2},
metrics={'output_1': 'metric_1', 'output_2': ['metric_2', 'metric_3']})
您可以在https://keras.io/getting-started/functional-api-guide/#multi-input-and-multi-output-models
中详细了解Keras的多输出模型。