Python sklearn recall_score用法及代码示例

本文简要介绍python语言中 sklearn.metrics.recall_score 的用法。

用法:

sklearn.metrics.recall_score(y_true, y_pred, *, labels=None, pos_label=1, average='binary', sample_weight=None, zero_division='warn')

计算召回率。

召回率是 tp / (tp + fn) 的比率，其中 tp 是真阳性数，fn 是假阴性数。召回率直观地是分类器找到所有正样本的能力。

最佳值为 1，最差值为 0。

在用户指南中阅读更多信息。

参数：

y_true：一维数组，或标签指示符数组/稀疏矩阵

基本事实(正确)目标值。

y_pred：一维数组，或标签指示符数组/稀疏矩阵

分类器返回的估计目标。

labels：类似数组，默认=无

当 average != 'binary' 时要包含的标签集，如果 average is None 则它们的顺序。可以排除数据中存在的标签，例如计算忽略多数负类的多类平均值，而数据中不存在的标签将导致宏观平均值中的 0 个分量。对于多标签目标，标签是列索引。默认情况下，y_true 和y_pred 中的所有标签都按排序顺序使用。

pos_label：str 或 int，默认 = 1

如果average='binary' 并且数据是二进制的，则要报告的类。如果数据是多类或多标签的，这将被忽略；设置 labels=[pos_label] 和 average != 'binary' 将仅报告该标签的分数。

average：{‘micro’, ‘macro’, ‘samples’, ‘weighted’, ‘binary’} 或无，默认='二进制'

多类/多标签目标需要此参数。如果 None ，则返回每个类的分数。否则，这将确定对数据执行的平均类型：

'binary'：

仅报告 pos_label 指定的类的结果。这仅适用于目标 (y_{true,pred}) 是二进制的。

'micro'：

通过计算总的真阳性、假阴性和假阳性来全局计算指标。

'macro'：

计算每个标签的指标，并找到它们的未加权平均值。这没有考虑标签不平衡。

'weighted'：

计算每个标签的指标，并通过支持度(每个标签的真实实例数)找到它们的平均加权值。这会改变 ‘macro’ 以解决标签不平衡问题；它可能导致 F-score 不在精确率和召回率之间。加权召回等于准确率。

'samples'：

计算每个实例的指标，并找到它们的平均值(仅对不同于 accuracy_score 的多标签分类有意义)。

sample_weight：形状类似数组 (n_samples,)，默认=None

样本权重。

zero_division：“warn”，0或1，默认=”warn”

设置零除法时要返回的值。如果设置为“warn”，这将作为 0，但也会引发警告。

返回：

recall：浮点数(如果平均值不是无)或形状的浮点数数组 (n_unique_labels,)

对二分类中正类的召回或多类任务中每个类召回的加权平均。

注意：

当 true positive + false negative == 0 时，召回返回 0 并引发 UndefinedMetricWarning 。可以使用 zero_division 修改此行为。

例子：

>>> from sklearn.metrics import recall_score
>>> y_true = [0, 1, 2, 0, 1, 2]
>>> y_pred = [0, 2, 1, 0, 0, 1]
>>> recall_score(y_true, y_pred, average='macro')
0.33...
>>> recall_score(y_true, y_pred, average='micro')
0.33...
>>> recall_score(y_true, y_pred, average='weighted')
0.33...
>>> recall_score(y_true, y_pred, average=None)
array([1., 0., 0.])
>>> y_true = [0, 0, 0, 0, 0, 0]
>>> recall_score(y_true, y_pred, average=None)
array([0.5, 0. , 0. ])
>>> recall_score(y_true, y_pred, average=None, zero_division=1)
array([0.5, 1. , 1. ])
>>> # multilabel classification
>>> y_true = [[0, 0, 0], [1, 1, 1], [0, 1, 1]]
>>> y_pred = [[0, 0, 0], [1, 1, 1], [1, 1, 0]]
>>> recall_score(y_true, y_pred, average=None)
array([1. , 1. , 0.5])