针对特定输入的多个指标

时间:2019-02-18 15:48:21

标签: python tensorflow keras

无论是自定义还是从keras导入,我都有多个损失和指标。有没有一种方法可以指定可以将哪些模型输出输入到哪个度量,而不是打印或计算所有模型输出?

1 个答案:

答案 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的多输出模型。