Python中ROC曲線繪制
首先以支持向量機模型為例
先導入需要使用的包，我們將使用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ù)劃分訓練集和測試集數(shù)據(jù)；
訓練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曲線的橫縱坐標，于是我們以變量fpr為橫坐標，tpr為縱坐標，繪制相應的ROC圖像如下：
值得注意的是上面的支持向量機模型使用的decision_function函數(shù)，是自己所特有的，而其他模型不能直接使用。
比如說我們想要使用其他模型（例如決策樹模型）的結果繪制ROC,直接套用上面的代碼，會報錯，會顯示沒有這個函數(shù)。
以決策樹模型為例，解決上述問題（適用于除向量機外的模型）
導入決策樹模型包以及訓練模型的代碼省略了，只需要手動改一改就行了，我們直接看繪圖的代碼！
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]就行了，讓我們看看結果：
可以看到哈，決策樹模型在這個數(shù)據(jù)集上的泛化能力不如支持向量機哈?。?！學廢了嗎。
我可以失敗，但絕不會認輸！
到此這篇關于Python中ROC曲線繪制的文章就介紹到這了,更多相關Python ROC曲線繪制內容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關文章希望大家以后多多支持腳本之家！
相關文章
這篇文章主要為大家詳細介紹了利用Python實現(xiàn)繪制3D立體愛心圖案的四種不同方法，文中的示例代碼講解詳細，感興趣的小伙伴可以動手嘗試一下2022-11-11
Python基于select實現(xiàn)的socket服務器
這篇文章主要介紹了Python基于select實現(xiàn)的socket服務器,實例分析了Python基于select與socket模塊實現(xiàn)socket通信的相關技巧,需要的朋友可以參考下2016-04-04
python?爬取豆瓣電影短評并利用wordcloud生成詞云圖
這篇文章主要介紹了python?爬取豆瓣電影短評并利用wordcloud生成詞云圖，文章圍繞主題展開詳細的內容介紹，具有一定的參考價值，需要的小伙伴可以參考一下2022-06-06
今天小編就為大家分享一篇淺談python中requests模塊導入的問題，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2018-05-05
這篇文章主要為大家詳細介紹了python實現(xiàn)滑雪者小游戲，文中示例代碼介紹的非常詳細，具有一定的參考價值，感興趣的小伙伴們可以參考一下2020-02-02
Python環(huán)境的安裝以及PyCharm編輯器配置教程詳解
優(yōu)質的教程可以讓我們少走很多彎路，這一點毋庸置疑。這篇文章主要為大家介紹了純凈Python環(huán)境的安裝以及PyCharm編輯器的配置，需要的可以參考一下2023-04-04
supervisor是一個用python語言編寫的進程管理工具,它可以很方便的監(jiān)聽、啟動、停止、重啟一個或多個進程,本文給大家介紹python進程管理工具supervisor安裝使用配置教程,感興趣的朋友一起看看吧2023-08-08
詳解利用python識別圖片中的條碼（pyzbar）及條碼圖片矯正和增強
這篇文章主要介紹了詳解利用python識別圖片中的條碼（pyzbar）及條碼圖片矯正和增強，文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值，需要的朋友們下面隨著小編來一起學習學習吧2020-11-11
Windows下python3安裝tkinter的問題及解決方法
這篇文章主要介紹了Windows下python3安裝tkinter問題及解決方法，本文給大家介紹的非常詳細，具有一定的參考借鑒價值,需要的朋友可以參考下2020-01-01
這篇文章主要為大家介紹了Python光學仿真wxpython之DC繪圖的基本概念及用法詳解，有需要的朋友可以借鑒參考下，希望能夠有所幫助2021-10-10