Tensorflow 中每个班级的 F1 分数指标

Posted 2023-02-16

【中文标题】Tensorflow 中每个班级的 F1 分数指标

【英文标题】：F1 Score metric per class in Tensorflow

【发布时间】：2021-10-06 07:32:28

【问题描述】：

我已经实施了以下指标来查看我认为相关的类的精度和召回率。

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 类，没有自定义函数。

【参考方案1】：

您可以查看 tensorflow-addons 包中的 https://www.tensorflow.org/addons/api_docs/python/tfa/metrics/F1Score。 具体来说，如果您需要每个班级的分数，则需要将 average 参数设置为 None 或 ma​​cro。