Python中ROC曲線怎么繪制

本篇內容介紹了“Python中ROC曲線怎么繪制”的有關知識，在實際案例的操作過程中，不少人都會遇到這樣的困境，接下來就讓小編帶領大家學習一下如何處理這些情況吧！希望大家仔細閱讀，能夠學有所成！

首先以支持向量機模型為例

先導入需要使用的包，我們將使用roc_curve這個函數繪制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函數，生成一個二分類的數據不平衡數據集；

使用train_test_split函數劃分訓練集和測試集數據；

訓練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函數生成三個變量，分別是fpr,tpr, thresholds，也就是假正例率（FPR）、真正例率（TPR）和閾值。

而其中的fpr,tpr正是我們繪制ROC曲線的橫縱坐標，于是我們以變量fpr為橫坐標，tpr為縱坐標，繪制相應的ROC圖像如下：

值得注意的是上面的支持向量機模型使用的decision_function函數，是自己所特有的，而其他模型不能直接使用。

比如說我們想要使用其他模型（例如決策樹模型）的結果繪制ROC,直接套用上面的代碼，會報錯，會顯示沒有這個函數。

以決策樹模型為例，解決上述問題（適用于除向量機外的模型）

導入決策樹模型包以及訓練模型的代碼省略了，只需要手動改一改就行了，我們直接看繪圖的代碼！

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')

可以看到我們直接把只適用于支持向量機模型的函數decision_function更改成predict_proba(X_test)[:,1]就行了，讓我們看看結果：

“Python中ROC曲線怎么繪制”的內容就介紹到這里了，感謝大家的閱讀。如果想了解更多行業相關的知識可以關注億速云網站，小編將為大家輸出更多高質量的實用文章！