Tensorflow中每类F1分数指标

Question

我已经实现了以下度量，以查看我认为相关的类的精确度和召回率

metrics=[tf.keras.metrics.Recall(class_id=1, name='Bkwd_R'),tf.keras.metrics.Recall(class_id=2, name='Fwd_R'),tf.keras.metrics.Precision(class_id=1, name='Bkwd_P'),tf.keras.metrics.Precision(class_id=2, name='Fwd_P')]

如何在Tensorflow 2.5中为F1分数实现相同的功能（即专门针对类1和类2，而不是类0，没有自定义函数）

Answer 1

您可以查看tensorflow插件包中的https://www.tensorflow.org/addons/api_docs/python/tfa/metrics/F1Score。 具体来说，如果需要每类分数，则需要将平均值参数设置为无，或宏