有时候模型有多个输出(比如一个分类问题一个回归问题), 需要定义不同的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']]
详细可以参考官方文档.
网友评论