Python sklearn fbeta_score用法及代码示例

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

用法:

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

计算F-beta 分数。

F-beta 分数是准确率和召回率的加权调和平均值，在 1 处达到最佳值，在 0 处达到最差值。

beta 参数确定组合分数中召回的权重。 beta < 1 赋予精度更多的权重，而beta > 1 有利于召回(beta -> 0 仅考虑精度，beta -> +inf 仅考虑召回)。

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

参数：

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

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

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

分类器返回的估计目标。

beta：浮点数

确定组合得分中召回的权重。

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，但也会引发警告。

返回：

fbeta_score：浮点数(如果平均值不是无)或浮点数数组，形状 = [n_unique_labels]

F-beta二进制分类中正类的分数或多类任务每个类的F-beta分数的加权平均值。

注意：

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

参考：

1

R.Baeza-Yates 和 B.Ribeiro-Neto(2011 年)。现代信息检索。艾迪生卫斯理，第 327-328 页。

2

Wikipedia entry for the F1-score 。

例子：

>>> from sklearn.metrics import fbeta_score
>>> y_true = [0, 1, 2, 0, 1, 2]
>>> y_pred = [0, 2, 1, 0, 0, 1]
>>> fbeta_score(y_true, y_pred, average='macro', beta=0.5)
0.23...
>>> fbeta_score(y_true, y_pred, average='micro', beta=0.5)
0.33...
>>> fbeta_score(y_true, y_pred, average='weighted', beta=0.5)
0.23...
>>> fbeta_score(y_true, y_pred, average=None, beta=0.5)
array([0.71..., 0.        , 0.        ])