如何解決ROC曲線畫出來只有一個點(diǎn)的問題

這篇文章主要介紹如何解決ROC曲線畫出來只有一個點(diǎn)的問題，文中介紹的非常詳細(xì)，具有一定的參考價值，感興趣的小伙伴們一定要看完！

之前在做kaggle比賽時，有個比賽使用AUC來評比的，當(dāng)時試著畫了ROC曲線，結(jié)果出來的下圖這樣的圖形。跟平時的ROC曲線差好遠(yuǎn)，就只有一個點(diǎn)。而別人家的都是很多轉(zhuǎn)折的，為啥我的不一樣。

我的圖如下：

正常的圖（sklearn上面截取的）：

思考過后，發(fā)現(xiàn)原來：

ROC曲線，一般適用于你的分類器輸出一個“概率值”，即這個樣本屬于某個類的概率是多少。 如此的話，你就需要設(shè)定一個閾值， 大于這個閾值屬于正類，小于這個閾值屬于負(fù)類。

從而，對于這個閾值P0， 就會得到對應(yīng)的TPR, FPR, 也就是ROC曲線上的一個點(diǎn)，你設(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曲線畫出來只有一個點(diǎn)的問題”這篇文章的所有內(nèi)容，感謝各位的閱讀！希望分享的內(nèi)容對大家有幫助，更多相關(guān)知識，歡迎關(guān)注億速云行業(yè)資訊頻道！