python代码画roc、auc曲线
时间: 2023-09-12 15:10:09 浏览: 115
好的,我可以回答这个问题。在Python中,可以使用sklearn中的ROC曲线函数和AUC函数来画ROC、AUC曲线。下面是一份简单的样例代码:
```
from sklearn.metrics import roc_curve, roc_auc_score
import matplotlib.pyplot as plt
import numpy as np
# 生成假数据
y_true = np.random.randint(0, 2, size=(100,))
y_scores = np.random.rand(100)
# 计算fpr和tpr
fpr, tpr, thresholds = roc_curve(y_true, y_scores)
# 计算AUC
auc_score = roc_auc_score(y_true, y_scores)
# 画ROC曲线
plt.plot(fpr, tpr, label='ROC curve (area = {:.2f})'.format(auc_score))
plt.plot([0, 1], [0, 1], linestyle='--', lw=2, color='r', alpha=.8)
plt.xlim([-0.05, 1.05])
plt.ylim([-0.05, 1.05])
plt.xlabel('False Positive Rate')
plt.ylabel('True Positive Rate')
plt.title('Receiver operating characteristic (ROC) curve')
plt.legend(loc="lower right")
plt.show()
```
希望这个代码能够帮助到你。
阅读全文