混淆矩阵绘制roc曲线
时间: 2024-01-04 12:20:44 浏览: 221
通过matlab绘制数据的ROC曲线,并输出TN,FP,FN,TP混淆矩阵,含仿真操作录像
5星 · 资源好评率100%
以下是绘制ROC曲线的步骤:
1. 导入所需的库:
```python
import matplotlib.pyplot as plt
```
2. 绘制混淆矩阵:
```python
# 假设混淆矩阵为:
# [[TN, FP],
# [FN, TP]]
confusion_matrix = [[TN, FP], [FN, TP]]
```
3. 计算真正例率(True Positive Rate,TPR)和假正例率(False Positive Rate,FPR):
```python
TPR = TP / (TP + FN)
FPR = FP / (FP + TN)
```
4. 绘制ROC曲线:
```python
plt.plot(FPR, TPR, color='darkorange', lw=2, label='ROC curve')
plt.plot([0, 1], [0, 1], color='navy', lw=2, linestyle='--')
plt.xlim([0.0, 1.0])
plt.ylim([0.0, 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()
```
阅读全文