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

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

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

语法格式

sklearn.metrics.auc(xy)

参数解释:

  • 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

 

posted @ 2023-04-29 00:00  yayagogogo  阅读(238)  评论(0编辑  收藏  举报