matthews_corrcoef#

sklearn.metrics.matthews_corrcoef(y_true, y_pred, *, sample_weight=None)[源码]#

计算马修斯相关系数 (MCC)。

马修斯相关系数在机器学习中用作衡量二分类和多分类质量的指标。它考虑了真阳性、假阳性、真阴性和假阴性,通常被认为是一种平衡的度量,即使类的尺寸差异很大也可以使用。MCC 本质上是一个介于 -1 和 +1 之间的相关系数值。+1 表示完美预测,0 表示平均随机预测,-1 表示反向预测。该统计量也称为 phi 系数。[来源:维基百科]

支持二分类和多分类标签。仅在二分类情况下,这与真阳性、假阳性和假阴性的信息相关。请参阅下面的参考资料。

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

参数:
y_true形状为 (n_samples,) 的类数组

真实(正确)的目标值。

y_pred形状为 (n_samples,) 的类数组

分类器返回的估计目标。

sample_weight形状为 (n_samples,) 的类数组,默认值为 None

样本权重。

版本 0.18 中新增。

返回:
mcc浮点数

马修斯相关系数(+1 表示完美预测,0 表示平均随机预测,-1 表示反向预测)。

参考文献

示例

>>> from sklearn.metrics import matthews_corrcoef
>>> y_true = [+1, +1, +1, -1]
>>> y_pred = [+1, -1, +1, +1]
>>> matthews_corrcoef(y_true, y_pred)
-0.33