MCC(Matthews Correlation Coefficient)和 F1-score 是两种常用的二分类模型评估指标。在 Python 中,可以使用 sklearn 库中的 metrics 模块计算 MCC 和 F1-score。

  1. 计算 MCC

MCC 的计算方法如下:

MCC公式

其中,TP 表示真正例(True Positive)的数量,TN 表示真负例(True Negative)的数量,FP 表示假正例(False Positive)的数量,FN 表示假负例(False Negative)的数量。

在 sklearn 库中,可以使用 metrics.matthews_corrcoef() 函数计算 MCC,示例代码如下:

from sklearn import metrics

y_true = [1, 1, 0, 0, 1, 0]
y_pred = [1, 0, 0, 1, 1, 0]

mcc = metrics.matthews_corrcoef(y_true, y_pred)
print(mcc)

输出结果为:

0.16666666666666666
  1. 计算 F1-score

F1-score 的计算方法如下:

F1-score公式

其中,precision 表示精确率(Precision),recall 表示召回率(Recall)。

在 sklearn 库中,可以使用 metrics.f1_score() 函数计算 F1-score,示例代码如下:

from sklearn import metrics

y_true = [1, 1, 0, 0, 1, 0]
y_pred = [1, 0, 0, 1, 1, 0]

f1_score = metrics.f1_score(y_true, y_pred)
print(f1_score)

输出结果为:

0.6666666666666666

注意:在使用这些函数时,需要保证 y_true 和 y_pred 的长度相等。

Python 计算 MCC 和 F1-score:二分类模型评估指标

原文地址: https://www.cveoy.top/t/topic/lQfX 著作权归作者所有。请勿转载和采集!

免费AI点我,无需注册和登录