如何利用scikitlearn畫ROC曲線-創(chuàng)新互聯(lián)

創(chuàng)新互聯(lián)www.cdcxhl.cn八線動態(tài)BGP香港云服務(wù)器提供商,新人活動買多久送多久,劃算不套路!

網(wǎng)站建設(shè)哪家好,找創(chuàng)新互聯(lián)!專注于網(wǎng)頁設(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)行模型評估,對于二分類來說,AUC,ROC這兩個(gè)指標(biāo)用到最多,所以 利用sklearn里面相應(yīng)的函數(shù)進(jìn)行模塊搭建。

具體實(shí)現(xiàn)的代碼可以參照下面博友的代碼,評估svm的分類指標(biāo)。注意里面的一些細(xì)節(jié)需要注意,一個(gè)是調(diào)用roc_curve 方法時(shí),指明目標(biāo)標(biāo)簽,否則會報(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)     #對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)) 

當(dāng)前名稱:如何利用scikitlearn畫ROC曲線-創(chuàng)新互聯(lián)
轉(zhuǎn)載來于:http://www.muchs.cn/article26/depijg.html

成都網(wǎng)站建設(shè)公司_創(chuàng)新互聯(lián),為您提供軟件開發(fā)網(wǎng)站導(dǎo)航、外貿(mào)建站、Google全網(wǎng)營銷推廣、網(wǎng)站建設(shè)

廣告

聲明:本網(wǎng)站發(fā)布的內(nèi)容(圖片、視頻和文字)以用戶投稿、用戶轉(zhuǎn)載內(nèi)容為主,如果涉及侵權(quán)請盡快告知,我們將會在第一時(shí)間刪除。文章觀點(diǎn)不代表本網(wǎng)站立場,如需處理請聯(lián)系客服。電話:028-86922220;郵箱:631063699@qq.com。內(nèi)容未經(jīng)允許不得轉(zhuǎn)載,或轉(zhuǎn)載時(shí)需注明來源: 創(chuàng)新互聯(lián)

成都定制網(wǎng)站網(wǎng)頁設(shè)計(jì)