roc_curve怎么使用
时间: 2024-01-31 10:03:02 浏览: 177
`roc_curve` 是一个用于绘制 ROC 曲线的函数,通常在 Python 中使用 `sklearn.metrics` 库中的 `roc_curve` 函数。
以下是使用 `roc_curve` 函数的基本步骤:
1. 首先,从模型中获取分类器的预测概率(或得分)和真实标签。
2. 使用 `roc_curve` 函数获取不同阈值下的 FPR、TPR 和阈值。
3. 绘制 ROC 曲线。
下面是一个使用 `roc_curve` 函数的简单示例:
``` python
from sklearn.metrics import roc_curve
import matplotlib.pyplot as plt
# 获取预测概率和真实标签
y_pred_prob = model.predict_proba(X_test)[:, 1]
y_true = y_test
# 计算 FPR、TPR 和阈值
fpr, tpr, thresholds = roc_curve(y_true, y_pred_prob)
# 绘制 ROC 曲线
plt.plot(fpr, tpr)
plt.xlabel('False Positive Rate')
plt.ylabel('True Positive Rate')
plt.title('ROC Curve')
plt.show()
```
在以上代码中,`model` 是模型对象,`X_test` 是测试集特征矩阵,`y_test` 是测试集标签。`y_pred_prob` 是模型的预测概率,`fpr`、`tpr` 和 `thresholds` 分别是不同阈值下的 FPR、TPR 和阈值。最后,使用 `plt.plot` 函数绘制 ROC 曲线即可。
阅读全文