解決ROC曲線畫出來只有一個(gè)點(diǎn)的問題
之前在做kaggle比賽時(shí)，有個(gè)比賽使用AUC來評比的，當(dāng)時(shí)試著畫了ROC曲線，結(jié)果出來的下圖這樣的圖形。跟平時(shí)的ROC曲線差好遠(yuǎn)，就只有一個(gè)點(diǎn)。而別人家的都是很多轉(zhuǎn)折的，為啥我的不一樣。
我的圖如下：
正常的圖（sklearn上面截取的）：
思考過后，發(fā)現(xiàn)原來：
ROC曲線，一般適用于你的分類器輸出一個(gè)“概率值”，即這個(gè)樣本屬于某個(gè)類的概率是多少。 如此的話，你就需要設(shè)定一個(gè)閾值， 大于這個(gè)閾值屬于正類，小于這個(gè)閾值屬于負(fù)類。
從而，對于這個(gè)閾值P0， 就會得到對應(yīng)的TPR, FPR, 也就是ROC曲線上的一個(gè)點(diǎn)，你設(shè)置不同的閾值，就會得到不同的TPR, FPR， 從而構(gòu)成ROC曲線。
通常來說 閾值降低，即進(jìn)入正類的門檻變低， TPR會變大，但是FPR也會變大， 看他們誰變的快。
之前畫線的時(shí)候直接用了分類的預(yù)測值，而沒有用上概率這玩意，被劃分為正類的概率。。修改下程序，求多一個(gè)概率就能畫出正確的圖形啦。
如果你用GDBT算法的時(shí)候：
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) ###畫圖的時(shí)候要用預(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曲線畫出來只有一個(gè)點(diǎn)的問題就是小編分享給大家的全部內(nèi)容了，希望能給大家一個(gè)參考，也希望大家多多支持腳本之家。
相關(guān)文章
Python打開指定網(wǎng)頁使用requests模塊爬蟲示例詳解
這篇文章主要介紹了Python打開指定網(wǎng)頁使用requests模塊爬蟲的示例,Python?requests是一個(gè)常用的HTTP請求庫,可以方便地向網(wǎng)站發(fā)送HTTP請求,并獲取響應(yīng)結(jié)果,requests模塊比urllib模塊更簡潔,感興趣的朋友可以參考下2024-02-02
這篇文章主要介紹了python代碼實(shí)現(xiàn)備忘錄案例講解,本篇文章通過簡要的案例,講解了該項(xiàng)技術(shù)的了解與使用,以下就是詳細(xì)內(nèi)容,需要的朋友可以參考下2021-07-07
Python實(shí)現(xiàn)新版正方系統(tǒng)滑動驗(yàn)證碼識別
這篇文章主要介紹了基于Python實(shí)現(xiàn)新版正方系統(tǒng)滑動驗(yàn)證碼識別算法和方案，文中示例代碼對我們的學(xué)習(xí)和工作有一定的幫助，感興趣的可以了解一下2021-12-12
sqlalchemy實(shí)現(xiàn)時(shí)間列自動更新教程
這篇文章主要介紹了sqlalchemy實(shí)現(xiàn)時(shí)間列自動更新教程，具有很好的參考價(jià)值，希望對大家有所幫助。一起跟隨小編過來看看吧2020-09-09
今天小編就為大家分享一篇利用Python正則表達(dá)式過濾敏感詞的方法，具有很好的參考價(jià)值，希望對大家有所幫助。一起跟隨小編過來看看吧2019-01-01
這篇文章主要介紹了Python中的random()方法的使用,是Python入門的基礎(chǔ)知識,需要的朋友可以參考下2015-05-05
Django網(wǎng)絡(luò)框架之HelloDjango項(xiàng)目創(chuàng)建教程
這篇文章主要介紹了Django網(wǎng)絡(luò)框架之HelloDjango項(xiàng)目創(chuàng)建,結(jié)合實(shí)例形式詳細(xì)分析了Django框架創(chuàng)建HelloDjango項(xiàng)目的具體步驟與詳細(xì)實(shí)現(xiàn)技巧,需要的朋友可以參考下2019-06-06
pandas.loc 選取指定列進(jìn)行操作的實(shí)例
今天小編就為大家分享一篇pandas.loc 選取指定列進(jìn)行操作的實(shí)例，具有很好的參考價(jià)值，希望對大家有所幫助。一起跟隨小編過來看看吧2018-05-05
Python利用matplotlib實(shí)現(xiàn)餅圖繪制
Pyplot作為Matplotlib的子庫，提供了和MATLAB差不多的繪圖API。因此Pyplot作為常用的繪圖模塊，能很方便讓用戶繪制2D圖表。本文將為大家介紹如何利用Matplotlib繪制餅圖，感興趣的小伙伴可以了解一下2021-12-12
Python Django使用forms來實(shí)現(xiàn)評論功能
這篇文章主要為大家詳細(xì)介紹了Python Django使用forms來實(shí)現(xiàn)評論功能，具有一定的參考價(jià)值，感興趣的小伙伴們可以參考一下2016-08-08