從10個不同的閾值創建ROC

Question

我有svmlight的輸出，其中x =預測（0.1，-0.6,1.2，-0.7 ...），y =實際的類{+ 1，-1}。我想爲10個特定的不同閾值創建ROC曲線（讓t爲包含10個不同閾值的矢量）。我檢查了ROCR軟件包，但我沒有看到提供閾值向量的任何選項。我需要爲每個閾值和繪圖計算TPR和FPR。有沒有其他方法可以做到這一點？我是R編程新手。

Answer 1

ROCR通過​​將TPR和FPR繪製成許多不同的閾值來創建ROC曲線。這可以通過一組預測和標籤來完成，因爲如果觀察被分類爲對於一個閾值爲正，則它也將在較低閾值處被分類爲正。我發現this paper有助於更詳細地解釋ROC曲線。

在ROCR以下，其中x是預測的矢量，y是類標籤的載體可以創建情節：

pred <- prediction(x,y) 
perf <- performance(pred,"tpr","fpr") 
plot(perf) 

如果你想訪問與相關聯的TPR和FPR所有閾值，可以檢查性能對象「PERF」：

str(perf) 

以下的答案必須展示如何獲得更詳細的臨界值：

https://stackoverflow.com/a/16347508/786220

Answer 2

您可以使用pROC包執行此操作。首先創建ROC曲線（對於所有閾值）：

myROC <- roc(y, x) # with the x and y you defined in your question 

然後查詢該曲線爲10（或任意數量的）閾值，你保存在t：

coords(myROC, x = t, input="threshold", ret = c("threshold", "se", "sp")) 

靈敏度是你的TPR，而特異性是您的FPR。

---

聲明：我是pROC的作者。