Tensorflow 中每个 class 的 F1 分数指标

F1 Score metric per class in Tensorflow

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

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 分数(即专门针对 class 1 和 class 2,而不是 class 0,没有自定义函数。

你可以在 tensorflow-addons 包中查看 https://www.tensorflow.org/addons/api_docs/python/tfa/metrics/F1Score。 具体来说,如果您需要每个 class 分数,则需要将 average 参数设置为 None,或者.