Tensorflow多分类指标如何用tf.metrics计算precision、recall、f1值

Yumi Mammy

于 2020-07-20 22:10:13 发布

阅读量2.6k

点赞数

CC 4.0 BY-SA版权

本文链接：https://blog.csdn.net/weixin_42234769/article/details/107475343

本文介绍如何在Tensorflow中使用tf.metrics计算多分类任务的precision、recall和F1值，参考自guillaumegenthial的tf_metrics库。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

来自https://github.com/guillaumegenthial/tf_metrics。

"""Multiclass"""

__author__ = "Guillaume Genthial"

import numpy as np
import tensorflow as tf
from tensorflow.python.ops.metrics_impl import _streaming_confusion_matrix


def precision(labels, predictions, num_classes, pos_indices=None,
              weights=None, average='micro'):
    """Multi-class precision metric for Tensorflow
    Parameters
    ----------
    labels : Tensor of tf.int32 or tf.int64
        The true labels
    predictions : Tensor