python画auc,python画ROC曲线并标记AUC

python画ROC曲线并标记AUC

python画ROC曲线并标记AUC

定义画图ROC函数

def roc(name): # 生成ROC曲线

plt.figure(dpi=150,figsize=(3,3))

plt.plot(fpr_test,tpr_test,'b:',label=f'测试集AUC={round(metrics.auc(fpr_test,tpr_test),4)}')

plt.plot(fpr_train,tpr_train,'r-.',label=f'训练集AUC={round(metrics.auc(fpr_train,tpr_train),4)}')

plt.legend();plt.title(f'ROC曲线-{name}');plt.xlabel('假正率');plt.ylabel('真正率')

# plt.grid()

plt.plot([0, 1], [0, 1], 'k--',linewidth = 0.5)

x2_train, x2_test, y2_train, y2_test = train_test_split(x1, y1, test_size=0.3,random_state=3)

from sklearn.linear_model import LogisticRegression as LR

model = LR()

model.fit(x2_train,y2_train)

# roc

fpr_test,tpr_test,th_test = metrics.roc_curve(y2_test,model.predict_proba(x2_test)[:,1]) # 测试集

fpr.append(fpr_test); tpr.append(tpr_test); auc.append(round(metrics.auc(fpr_test,tpr_test),4))

fpr_train,tpr_train,th_train = metrics.roc_curve(y2_train,model.predict_proba(x2_train)[:,1]) # 训练集

roc('逻辑回归')

python画ROC曲线并标记AUC相关教程