python画auc_利用python画出AUC曲线的实例

以load_breast_cancer数据集为例，模型细节不重要，重点是画AUC的代码。

直接上代码： from sklearn.datasets import load_breast_cancer

from sklearn import metrics

from sklearn.ensemble import RandomForestClassifier

from sklearn.model_selection import train_test_split

import pylab as plt

import warnings;warnings.filterwarnings('ignore')

dataset = load_breast_cancer()

data = dataset.data

target = dataset.target

X_train,X_test,y_train,y_test = train_test_split(data,target,test_size=0.2)

rf = RandomForestClassifier(n_estimators=5)

rf.fit(X_train,y_train)

pred = rf.predict_proba(X_test)[:,1]

#############画图部分

fpr, tpr, threshold = metrics.roc_curve(y_test, pred)

roc_auc = metrics.auc(fpr, tpr)

plt.figure(figsize=(6,6))

plt.title('Validation ROC')