解決ROC曲線畫出來只有一個點的問題
之前在做kaggle比賽時,有個比賽使用AUC來評比的,當(dāng)時試著畫了ROC曲線,結(jié)果出來的下圖這樣的圖形。跟平時的ROC曲線差好遠(yuǎn),就只有一個點。而別人家的都是很多轉(zhuǎn)折的,為啥我的不一樣。
我的圖如下:
正常的圖(sklearn上面截取的):
思考過后,發(fā)現(xiàn)原來:
ROC曲線,一般適用于你的分類器輸出一個“概率值”,即這個樣本屬于某個類的概率是多少。 如此的話,你就需要設(shè)定一個閾值, 大于這個閾值屬于正類,小于這個閾值屬于負(fù)類。
從而,對于這個閾值P0, 就會得到對應(yīng)的TPR, FPR, 也就是ROC曲線上的一個點,你設(shè)置不同的閾值,就會得到不同的TPR, FPR, 從而構(gòu)成ROC曲線。
通常來說 閾值降低,即進(jìn)入正類的門檻變低, TPR會變大,但是FPR也會變大, 看他們誰變的快。
之前畫線的時候直接用了分類的預(yù)測值,而沒有用上概率這玩意,被劃分為正類的概率。。修改下程序,求多一個概率就能畫出正確的圖形啦。
如果你用GDBT算法的時候:
gbc = GradientBoostingClassifier() gbc.fit(x_train, y_train) resu = gbc.predict(x_test) #進(jìn)行預(yù)測 y_pred_gbc = gbc.predict_proba(x_test)[:,1] ###這玩意就是預(yù)測概率的 fpr, tpr, threshold = roc_curve(y_test, y_pred_gbc) ###畫圖的時候要用預(yù)測的概率,而不是你的預(yù)測的值 plt.plot(fpr, tpr, 'b', label='AUC = %0.2f' % rocauc)#生成ROC曲線 plt.legend(loc='lower right') plt.plot([0, 1], [0, 1], 'r--') plt.xlim([0, 1]) plt.ylim([0, 1]) plt.ylabel('真正率') plt.xlabel('假正率') plt.show()
以上這篇解決ROC曲線畫出來只有一個點的問題就是小編分享給大家的全部內(nèi)容了,希望能給大家一個參考,也希望大家多多支持腳本之家。
相關(guān)文章
Python打開指定網(wǎng)頁使用requests模塊爬蟲示例詳解
這篇文章主要介紹了Python打開指定網(wǎng)頁使用requests模塊爬蟲的示例,Python?requests是一個常用的HTTP請求庫,可以方便地向網(wǎng)站發(fā)送HTTP請求,并獲取響應(yīng)結(jié)果,requests模塊比urllib模塊更簡潔,感興趣的朋友可以參考下2024-02-02Python實現(xiàn)新版正方系統(tǒng)滑動驗證碼識別
這篇文章主要介紹了基于Python實現(xiàn)新版正方系統(tǒng)滑動驗證碼識別算法和方案,文中示例代碼對我們的學(xué)習(xí)和工作有一定的幫助,感興趣的可以了解一下2021-12-12Django網(wǎng)絡(luò)框架之HelloDjango項目創(chuàng)建教程
這篇文章主要介紹了Django網(wǎng)絡(luò)框架之HelloDjango項目創(chuàng)建,結(jié)合實例形式詳細(xì)分析了Django框架創(chuàng)建HelloDjango項目的具體步驟與詳細(xì)實現(xiàn)技巧,需要的朋友可以參考下2019-06-06Python利用matplotlib實現(xiàn)餅圖繪制
Pyplot作為Matplotlib的子庫,提供了和MATLAB差不多的繪圖API。因此Pyplot作為常用的繪圖模塊,能很方便讓用戶繪制2D圖表。本文將為大家介紹如何利用Matplotlib繪制餅圖,感興趣的小伙伴可以了解一下2021-12-12Python Django使用forms來實現(xiàn)評論功能
這篇文章主要為大家詳細(xì)介紹了Python Django使用forms來實現(xiàn)評論功能,具有一定的參考價值,感興趣的小伙伴們可以參考一下2016-08-08