利用python畫出AUC曲線的實(shí)例
以load_breast_cancer數(shù)據(jù)集為例,模型細(xì)節(jié)不重要,重點(diǎn)是畫AUC的代碼。
直接上代碼:
from sklearn.datasets import load_breast_cancerfrom sklearn import metricsfrom sklearn.ensemble import RandomForestClassifierfrom sklearn.model_selection import train_test_splitimport pylab as pltimport warnings;warnings.filterwarnings(’ignore’)dataset = load_breast_cancer()data = dataset.datatarget = dataset.targetX_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’)plt.plot(fpr, tpr, ’b’, label = ’Val AUC = %0.3f’ % roc_auc)plt.legend(loc = ’lower right’)plt.plot([0, 1], [0, 1],’r--’)plt.xlim([0, 1])plt.ylim([0, 1])plt.ylabel(’True Positive Rate’)plt.xlabel(’False Positive Rate’)plt.show()
補(bǔ)充拓展:Python機(jī)器學(xué)習(xí)中的roc_auc曲線繪制
廢話不多說,直接上代碼
from sklearn.metrics import roc_curve,aucfrom sklearn.ensemble import RandomForestClassifierimport matplotlib.pyplot as pltfrom sklearn.model_selection import train_test_splitx_train,y_train,x_test,y_test=train_test_split(x,y,test_size=0.2)rf=RandomForestClassifier()rf.fit(x_train,y_train)rf.score(x_train,y_train)print(’trainscore:’+str(rfbest.score(x_train,y_train)))print(’testscore:’+str(rfbest.score(x_test,y_test)))y_score=rfbest.fit(x_train,y_train).predict_proba(x_test) #descision_function()不可用print(type(y_score))fpr,tpr,threshold=roc_curve(y_test,y_score[:, 1])roc_auc=auc(fpr,tpr)plt.figure(figsize=(10,10))plt.plot(fpr, tpr, color=’darkorange’,lw=2, label=’ROC curve (area = %0.2f)’ % roc_auc) ###假正率為橫坐標(biāo),真正率為縱坐標(biāo)做曲線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 example’)plt.legend(loc='lower right')plt.show()
以上這篇利用python畫出AUC曲線的實(shí)例就是小編分享給大家的全部?jī)?nèi)容了,希望能給大家一個(gè)參考,也希望大家多多支持好吧啦網(wǎng)。
相關(guān)文章:
1. 告別AJAX實(shí)現(xiàn)無刷新提交表單2. 使用XSL將XML文檔中的CDATA注釋輸出為HTML文本3. PHP循環(huán)與分支知識(shí)點(diǎn)梳理4. JavaWeb Servlet中url-pattern的使用5. ASP 信息提示函數(shù)并作返回或者轉(zhuǎn)向6. 讀寫xml文件的2個(gè)小函數(shù)7. 低版本IE正常運(yùn)行HTML5+CSS3網(wǎng)站的3種解決方案8. jsp實(shí)現(xiàn)簡(jiǎn)單用戶7天內(nèi)免登錄9. chat.asp聊天程序的編寫方法10. XML入門的常見問題(一)
