ROC曲线学习

from sklearn.svm import SVC
from sklearn.metrics import confusion_matrix
from sklearn import metrics
import matplotlib.pyplot as plt

def svm_func(x_train,y_train,x_test,y_test):
    clf = SVC(kernel = ‘linear‘, verbose=True)
    clf.fit(x_train,y_train)
    pred = clf.predict(x_test)

    #混淆矩阵， labels为数据中各类标签值，这里0为负样本，2为正样本
    cm = confusion_matrix(y_test, pred, labels=[0, 2]) #纵坐标是y_test，横坐标是pred
    print(cm)

    # 绘制ROC曲线
    fpr, tpr, thresholds = metrics.roc_curve(y_test, pred, pos_label=2) # pos_label表示选定该标签是正，其他的均为负，绘制ROC曲线
    auc = metrics.auc(fpr, tpr)  #计算AUC值
    
    # # 设置绘图风格
    plt.figure()
    plt.style.use(‘ggplot‘)
    
    # 绘制ROC曲线
    plt.plot(fpr, tpr,‘‘)
    # 绘制参考线
    plt.plot((0,1),(0,1),‘r--‘)
    # 添加文本
    plt.text(0.5,0.5,‘AUC值为%.2f‘ % auc, fontsize=16)

    plt.title(‘支持向量机模型的ROC曲线‘)
    plt.xlabel(‘1-specificity‘)
    plt.ylabel(‘Sensitivity‘)
    
    # 中文和负号的正常显示
    plt.rcParams[‘font.sans-serif‘] = [‘SimHei‘]
    plt.rcParams[‘axes.unicode_minus‘] = False
    
    # 去除图形顶部边界和右边界的刻度
    plt.tick_params(top=False, right=False)
    plt.show()