創(chuàng)新互聯(lián)www.cdcxhl.cn八線動(dòng)態(tài)BGP香港云服務(wù)器提供商,新人活動(dòng)買多久送多久,劃算不套路!
網(wǎng)站建設(shè)哪家好,找創(chuàng)新互聯(lián)!專注于網(wǎng)頁(yè)設(shè)計(jì)、網(wǎng)站建設(shè)、微信開發(fā)、成都小程序開發(fā)、集團(tuán)企業(yè)網(wǎng)站建設(shè)等服務(wù)項(xiàng)目。為回饋新老客戶創(chuàng)新互聯(lián)還提供了松桃免費(fèi)建站歡迎大家使用!小編給大家分享一下如何利用scikitlearn畫ROC曲線,希望大家閱讀完這篇文章后大所收獲,下面讓我們一起去探討方法吧!
一個(gè)完整的數(shù)據(jù)挖掘模型,最后都要進(jìn)行模型評(píng)估,對(duì)于二分類來說,AUC,ROC這兩個(gè)指標(biāo)用到最多,所以 利用sklearn里面相應(yīng)的函數(shù)進(jìn)行模塊搭建。
具體實(shí)現(xiàn)的代碼可以參照下面博友的代碼,評(píng)估svm的分類指標(biāo)。注意里面的一些細(xì)節(jié)需要注意,一個(gè)是調(diào)用roc_curve 方法時(shí),指明目標(biāo)標(biāo)簽,否則會(huì)報(bào)錯(cuò)。
具體是這個(gè)參數(shù)的設(shè)置pos_label ,以前在unionbigdata實(shí)習(xí)時(shí)學(xué)到的。
重點(diǎn)是以下的代碼需要根據(jù)實(shí)際改寫:
mean_tpr = 0.0 mean_fpr = np.linspace(0, 1, 100) all_tpr = [] y_target = np.r_[train_y,test_y] cv = StratifiedKFold(y_target, n_folds=6) #畫ROC曲線和計(jì)算AUC fpr, tpr, thresholds = roc_curve(test_y, predict,pos_label = 2)##指定正例標(biāo)簽,pos_label = ###########在數(shù)之聯(lián)的時(shí)候?qū)W到的,要制定正例 mean_tpr += interp(mean_fpr, fpr, tpr) #對(duì)mean_tpr在mean_fpr處進(jìn)行插值,通過scipy包調(diào)用interp()函數(shù) mean_tpr[0] = 0.0 #初始處為0 roc_auc = auc(fpr, tpr) #畫圖,只需要plt.plot(fpr,tpr),變量roc_auc只是記錄auc的值,通過auc()函數(shù)能計(jì)算出來 plt.plot(fpr, tpr, lw=1, label='ROC %s (area = %0.3f)' % (classifier, roc_auc))