Python中ROC曲線繪制
首先以支持向量機模型為例
先導(dǎo)入需要使用的包，我們將使用roc_curve這個函數(shù)繪制ROC曲線！
from sklearn.svm import SVC from sklearn.metrics import roc_curve from sklearn.datasets import make_blobs from sklearn. model_selection import train_test_split import matplotlib.pyplot as plt %matplotlib inline
然后使用下面make_blobs函數(shù)，生成一個二分類的數(shù)據(jù)不平衡數(shù)據(jù)集；
使用train_test_split函數(shù)劃分訓(xùn)練集和測試集數(shù)據(jù)；
訓(xùn)練SVC模型。
X,y = make_blobs(n_samples=(4000,500), cluster_std=[7,2], random_state=0) X_train,X_test,y_train, y_test = train_test_split(X,y,random_state=0) clf = SVC(gamma=0.05).fit(X_train, y_train)
fpr,tpr, thresholds = roc_curve(y_test,clf.decision_function(X_test))
plt.plot(fpr,tpr,label='ROC')
plt.xlabel('FPR')
plt.ylabel('TPR')
從上面的代碼可以看到，我們使用roc_curve函數(shù)生成三個變量，分別是fpr,tpr, thresholds，也就是假正例率（FPR）、真正例率（TPR）和閾值。
而其中的fpr,tpr正是我們繪制ROC曲線的橫縱坐標(biāo)，于是我們以變量fpr為橫坐標(biāo)，tpr為縱坐標(biāo)，繪制相應(yīng)的ROC圖像如下：
值得注意的是上面的支持向量機模型使用的decision_function函數(shù)，是自己所特有的，而其他模型不能直接使用。
比如說我們想要使用其他模型（例如決策樹模型）的結(jié)果繪制ROC,直接套用上面的代碼，會報錯，會顯示沒有這個函數(shù)。
以決策樹模型為例，解決上述問題（適用于除向量機外的模型）
導(dǎo)入決策樹模型包以及訓(xùn)練模型的代碼省略了，只需要手動改一改就行了，我們直接看繪圖的代碼！
fpr,tpr, thresholds = roc_curve(y_test,clf.predict_proba(X_test)[:,1])
plt.plot(fpr,tpr,label='ROC')
plt.xlabel('FPR')
plt.ylabel('TPR')
可以看到我們直接把只適用于支持向量機模型的函數(shù)decision_function更改成predict_proba(X_test)[:,1]就行了，讓我們看看結(jié)果：
可以看到哈，決策樹模型在這個數(shù)據(jù)集上的泛化能力不如支持向量機哈?。?！學(xué)廢了嗎。
我可以失敗，但絕不會認(rèn)輸！
到此這篇關(guān)于Python中ROC曲線繪制的文章就介紹到這了,更多相關(guān)Python ROC曲線繪制內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！
相關(guān)文章
這篇文章主要介紹了python類屬性學(xué)習(xí)深入講解，文中對于python的類屬性的理解有正在學(xué)習(xí)python的同學(xué)可以一塊學(xué)習(xí)下2021-03-03
利用pandas如何實現(xiàn)類似sql中的left join操作
這篇文章主要介紹了利用pandas如何實現(xiàn)類似sql中的left join操作，具有很好的參考價值，希望對大家有所幫助。如有錯誤或未考慮完全的地方，望不吝賜教2023-06-06
這篇文章主要介紹了Python爬蟲實現(xiàn)驗證碼登錄，文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2019-05-05
今天小編就為大家分享一篇python寫入并獲取剪切板內(nèi)容的實例，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2018-05-05
Python輸出列表(List)不帶中括號和引號的問題及解決方法
這篇文章主要介紹了Python輸出列表(List)不帶中括號和引號的問題及解決方法，本文給大家介紹的非常詳細(xì)，對大家的學(xué)習(xí)或工作具有一定的參考借鑒價值,需要的朋友可以參考下2023-02-02
這篇文章主要介紹了Django的forms組件，本文通過實例代碼介紹了Django的forms組件,需要的朋友可以參考下2018-11-11
在python新式類中，可以定義一個變量__slots__，它的作用是阻止在實例化類時為實例分配dict，下面這篇文章主要給大家介紹了關(guān)于Python中__slots__的相關(guān)資料，需要的朋友可以參考借鑒，下面來一起看看吧。2017-07-07
python網(wǎng)絡(luò)爬蟲學(xué)習(xí)筆記（1）
這篇文章主要為大家詳細(xì)介紹了python網(wǎng)絡(luò)爬蟲學(xué)習(xí)筆記的第一篇，具有一定的參考價值，感興趣的小伙伴們可以參考一下2018-04-04
Python中使用haystack實現(xiàn)django全文檢索搜索引擎功能
django是python語言的一個web框架，功能強大。配合一些插件可為web網(wǎng)站很方便地添加搜索功能。下面通過本文給大家分享Python中使用haystack實現(xiàn)django全文檢索搜索引擎功能，感興趣的朋友一起看看吧2017-08-08
今天小編就為大家分享一篇使用python的pyplot繪制函數(shù)實例，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2020-02-02