Keras训练神经网络，训练过程中添加召回率、F1等多元评价指标

问题描述

使用keras训练神经网络，常规情况下在model.compile通过metrics定义accuracy作为评价指标

model.compile(loss='categorical_crossentropy',
                  optimizer='rmsprop',
                  metrics=['accuracy'])

在训练过程中仅显示- loss、 - acc、- val_loss和- val_acc
但在实际任务中，希望可以记录召回率、F1等其它评价指标

使用的库

使用keras_metrics库解决这一问题。安装方式为：

pip install keras_metrics

深入了解，自行前往源码地址 https://github.com/netrack/keras-metrics

实例

导入keras_metrics库

import keras_metrics

在model.compile通过metrics直接添加

model.compile(loss='categorical_crossentropy',
                  optimizer='rmsprop',
                  metrics=['accuracy',
                 		    keras_metrics.f1_score(),
                		    keras_metrics.precision(), 
                		    keras_metrics.recall()
                		  ]
              )

训练中实际效果：

妥了。