R

使用 caret 包是否可以獲得特定閾值的混淆矩陣?

  • August 7, 2014

我已經獲得了二元響應的邏輯回歸模型(通過),並且通過intrain獲得了邏輯混淆矩陣。它給了我邏輯模型混淆矩陣,儘管我不確定使用什麼閾值來獲得它。如何使用in獲得特定閾值的混淆矩陣?confusionMatrix``caret``confusionMatrix``caret

R 中的大多數分類模型都會產生類預測和每個類的概率。對於二進制數據,幾乎在每種情況下,類別預測都基於 50% 的概率截止值。

glm是一樣的。使用caret, usingpredict(object, newdata)為您提供預測的類別,並predict(object, new data, type = "prob")為您提供特定類別的概率(何時object由 生成train)。

您可以通過定義自己的模型並應用您想要的任何截止值來做不同的事情。該caret 網站還有一個使用重採樣來優化概率截止的示例。

tl;博士

confusionMatrix使用預測的類別,因此有 50% 的概率截止

最大限度

引用自:https://stats.stackexchange.com/questions/110969

comments powered by Disqus