2021SC@SDUSC
代码地址:similarity/f1_score.py at master · tensorflow/similarity · GitHub
import tensorflow as tf
from tensorflow_similarity.types import FloatTensor
from .classification_metric import ClassificationMetric
class F1Score(ClassificationMetric):
"""Calculates the harmonic mean of precision and recall.
Computes the F-1 Score given the query classification counts. The metric is
computed as follows:
$$
F_1 = 2 \cdot \frac{\textrm{precision} \cdot \textrm{recall}}{\textrm{precision} + \textrm{recall}}
$$
args:
name: Name associated with a specific metric object, e.g.,
f1@0.1
Usage with `tf.similarity.models.SimilarityModel()`:
```python
model.calibra