526互联

sklearn.metrics.auc—使用梯形规则计算曲线下面积(AUC)

发布时间 2023-04-29 00:07:45作者: yayagogogo

参考：https://scikit-learn.org/stable/modules/generated/sklearn.metrics.auc.html

在分类模型的性能评估指标总结章节中，我们讲到AUC曲线是分类模型的性能评价指标之一。接下来将对sklearn库中AUC曲线的具体计算方式进行讲解。

语法格式

sklearn.metrics.auc(x, y)

参数解释：

  • x: x坐标，必须单调递增或递减。
  • y: y坐标。

返回曲线下面积。

代码示例

import numpy as np
from sklearn.metrics import roc_curve, auc

y_test = np.array([1,1,0, 1,1])
y_pre = np.array([1,1,0,1,0])

# Compute ROC
fpr, tpr, thresholds = roc_curve(y_test,y_pre)
roc_auc = auc(fpr, tpr)
print(roc_auc) #0.875