代码详解:
import numpy as np
import matplotlib.pyplot as plt
from sklearn.preprocessing import label_binarize
from sklearn.metrics import confusion_matrix,classification_report
from sklearn.metrics import roc_curve, auc
from sklearn.metrics import cohen_kappa_score, accuracy_score
with open("batch_16_epoch_2.txt","r",encoding="utf-8") as f:
y_true, y_sore,y_pre=[],[],[]
f=f.readlines()
data=[i.split("\n")[0].split(" ") for i in f ]
print("# 第一列是真实值 第二列是分数 第三列是预测值",data)
for line in data:
y_true.append(int(line[0]))
y_sore.append(float(line[1]))
y_pre.append(int(line[2]))
from sklearn.metrics import roc_curve, auc
fpr, tpr, thresholds = roc_curve(y_true,y_pre)# 真实值 预测值
roc_auc = auc(fpr, tpr)
plt.title('Receiver Operating Characteristic')
plt.plot(fpr, tpr, '#9400D3',label=u'AUC = %0.3f'
Original: https://blog.csdn.net/weixin_55771290/article/details/126189762
Author: 数学是算法的灵魂
Title: python 针对二分类问题画roc曲线
原创文章受到原创版权保护。转载请注明出处:https://www.johngo689.com/666197/
转载文章受原作者版权保护。转载请注明原作者出处!