在Python中roc_auc_score函数如何计算ROC曲线下面积大小
时间: 2024-05-06 14:15:09 浏览: 88
曲线绘制_roc_ROC曲线绘制_PRC_auc曲线绘制_
5星 · 资源好评率100%
在Python中,可以使用`sklearn.metrics`模块中的`roc_auc_score`函数计算ROC曲线下面积大小。
具体来说,该函数接受两个参数:真实标签(y_true)和预测概率(y_score),并返回ROC曲线下面积大小。
下面是一个示例代码:
```python
from sklearn.metrics import roc_auc_score
y_true = [0, 1, 0, 1]
y_score = [0.1, 0.9, 0.3, 0.7]
auc = roc_auc_score(y_true, y_score)
print(auc)
```
输出结果为:
```
0.75
```
这表示ROC曲线下面积大小为0.75。
阅读全文