美文网首页
Keras之为模型不同的输出定义不同的metrics

Keras之为模型不同的输出定义不同的metrics

作者: Qiangest | 来源:发表于2021-07-05 20:02 被阅读0次

有时候模型有多个输出(比如一个分类问题一个回归问题), 需要定义不同的metrics(比如accuracy和mae).
我们知道Keras可以定义多个loss, 用list传入即可. 那么Metrics呢? 自然也是可以的.
有以下两种方法:

dictionary

比如

 metrics={'output_a': 'accuracy', 'output_b': ['accuracy', 'mse']}

其中 output_a是output的name.

list of metrics

比如

 metrics=[['accuracy'], ['accuracy', 'mse']] or metrics=['accuracy', ['accuracy', 'mse']]

详细可以参考官方文档.

相关文章

网友评论

      本文标题:Keras之为模型不同的输出定义不同的metrics

      本文链接:https://www.haomeiwen.com/subject/urtnultx.html