Python中的ROC曲線技巧-Python教學-PHP中文網

Python中的ROC曲線技巧

WBOY

發布： 2023-06-11 12:12:27

原創

1997 人瀏覽過

近年來，資料科學和機器學習的發展促使Python程式語言成為了資料科學家和機器學習工程師們最喜愛的語言之一。 Python提供了龐大的函式庫和框架，如NumPy、Pandas、Scikit-learn等，使得建立高品質的機器學習模型變得更加容易。

ROC曲線（接受者操作特徵曲線）是機器學習中的重要工具，它用於評估分類演算法的效能，並幫助我們了解演算法的分類能力和穩健性。在Python中，我們可以使用多種方式繪製ROC曲線來幫助評估分類演算法。本文將介紹Python中的ROC曲線繪製技巧，並示範如何使用Scikit-learn庫和Matplotlib庫來建立一張漂亮的ROC曲線。

ROC曲線是如何運作的？

ROC曲線是二元分類器效能評估中最常用的工具之一。此曲線透過繪製假陽性率（False Positive Rate）和真陽性率（True Positive Rate）之間的關係來說明分類器的性能。假陽性率是假正類佔所有負類的比例，真陽性率是真正類別佔所有正類的比例。 ROC曲線的X軸是假陽性率，而Y軸是真陽性率。

通常，在分類問題中涉及到一個二元判斷問題，其中正例和負例分別稱為“1”和“0”，分類器可以根據一定的閾值將實例分類為正例或負例。如果分類器的閾值過高，會導致大量實例被錯分為負例。這會增加假陰性率（False Negative Rate），可能導致分類器漏掉實例。相反，如果分類器的閾值過低，則會導致大量實例被分為正例，增加假陽性率，可能導致誤判。為了實現最佳的分類器，我們需要權衡這兩個錯誤類型。

一個理想的ROC曲線由真正率等於1和假正率等於0的點開始，在該點處，閾值被設定為最大值。當我們增加閾值時，真正率保持不變，但假正率會增加。因此，在ROC曲線上的任何點，與較高的假陽性率相比，較高的真陽性率和低的假陽性率都被認為是更好的效能表現。

ROC曲線的技巧

Python中有幾個繪製ROC曲線的技巧。以下是一些常用技巧：

使用Scikit-learn函式庫

#Scikit-learn提供了方便的函數來計算設定不同閾值下的真假陽性輸出，並返回假陽性率和真陽性率結果。一旦我們有了這些輸出，我們可以將它們視覺化為ROC曲線。以下是使用Scikit-learn函式庫計算並繪製ROC曲線的範例：

from sklearn.metrics import roc_curve
from sklearn.metrics import auc

fpr, tpr, thresholds = roc_curve(y_test, y_pred_prob)
roc_auc = auc(fpr, tpr)

plt.figure()
plt.plot(fpr, tpr, color='darkorange', lw=2, label='ROC curve (area = %0.2f)' % roc_auc)
plt.plot([0, 1], [0, 1], color='navy', lw=2, linestyle='--')
plt.xlim([0.0, 1.0])
plt.ylim([0.0, 1.05])
plt.xlabel('False Positive Rate')
plt.ylabel('True Positive Rate')
plt.title('Receiver Operating Characteristic (ROC) Curve')
plt.legend(loc="lower right")
plt.show()

登入後複製

在這個範例中，我們假設我們已經擬合了一個二元分類器，並使用測試集計算了機率。 y_test是測試資料的分類標籤，y_pred_prob是分類器預測出的機率。此範例計算了fpr和tpr，並使用Scikit-learn裡的auc函數計算ROC曲線下的面積，我們可以用Matplotlib畫出ROC曲線圖。圖將真陽性率繪製在Y軸上，將假陽性率繪製在X軸上。

使用Matplotlib庫自訂繪製

如果您想更自訂ROC曲線的外觀，那麼您可以使用Matplotlib來建立自己的圖表。以下是一個範例，展示如何利用Matplotlib來繪製ROC曲線：

import numpy as np
import matplotlib.pyplot as plt

# Generate some data
N      = 50
x_true = np.random.randn(N)
x_false= np.random.randn(N)

# Add some noise
x_true = x_true + np.random.randn(N) * 0.3
x_false= x_false + np.random.randn(N) * 0.3

# Create labels and predictions
y_true = np.ones(N)
y_false= np.zeros(N)
y_pred = np.concatenate([x_true, x_false])
y_true = np.concatenate([y_true, y_false])

# Determine threshold for each point
thresholds = np.sort(y_pred)
tpr_all    = []
fpr_all    = []
for threshold in thresholds:
    y_pred_bin = (y_pred >= threshold).astype(int)
    tn, fp, fn, tp = confusion_matrix(y_true, y_pred_bin).ravel()
    tpr = tp / (tp + fn)
    fpr = fp / (fp + tn)
    tpr_all.append(tpr)
    fpr_all.append(fpr)    

plt.figure()
plt.plot(fpr_all, tpr_all)
plt.plot([0, 1], [0, 1], '--', color='grey')
plt.xlabel("False Positive Rate")
plt.ylabel("True Positive Rate")
plt.title("ROC Curve")
plt.show()

登入後複製

在這個例子中，我們首先產生了一些模擬數據，然後透過添加一些雜訊使其更加現實。接下來，我們在組合的數據上創建了標籤和預測，並計算出每個點的真假陽性率和閾值。我們最後使用Matplotlib來繪製ROC曲線。這個例子說明如何用Python程式實作繪製ROC曲線，同時也展示了繪製自訂圖表的方法。

結論

ROC曲線是評估分類器效能的重要工具。在Python中，使用Scikit-learn和Matplotlib兩個函式庫都可以繪製ROC曲線。 Scikit-learn提供了方便的函數計算ROC曲線，而Matplotlib則提供了高度可自訂化的繪圖工具。概述的範例說明了繪製ROC曲線的兩種技術，無論您是否打算使用函式庫函數或自訂繪圖，這些技巧都有助於評估分類器在真實資料上的表現表現。

以上是Python中的ROC曲線技巧的詳細內容。更多資訊請關注PHP中文網其他相關文章！