evaluate.py 文件源码

def metrics_from_counts(counts: List[int]) -> Tuple[float, float, float, float]:
    """
    Computes classifier metrics given counts of correct, incorrect, missing and spurious

    :param counts: A (4,) vector of (correct, incorrect, missing, spurious)
    :return: acc, recall, precision and f1
    """

    eps = 1e-16
    correct, incorrect, missing, spurious = counts

    acc = correct / (correct + incorrect + missing + spurious + eps)
    recall = correct / (correct + incorrect + missing + eps)
    precision = correct / (correct + incorrect + spurious + eps)
    f1 = 2 * (precision * recall) / (recall + precision + eps)

    return acc, recall, precision, f1