R
使用 caret 包是否可以獲得特定閾值的混淆矩陣?
我已經獲得了二元響應的邏輯回歸模型(通過),並且通過in
train
獲得了邏輯混淆矩陣。它給了我邏輯模型混淆矩陣,儘管我不確定使用什麼閾值來獲得它。如何使用in獲得特定閾值的混淆矩陣?confusionMatrix``caret``confusionMatrix``caret
R 中的大多數分類模型都會產生類預測和每個類的概率。對於二進制數據,幾乎在每種情況下,類別預測都基於 50% 的概率截止值。
glm
是一樣的。使用caret
, usingpredict(object, newdata)
為您提供預測的類別,並predict(object, new data, type = "prob")
為您提供特定類別的概率(何時object
由 生成train
)。您可以通過定義自己的模型並應用您想要的任何截止值來做不同的事情。該
caret
網站還有一個使用重採樣來優化概率截止的示例。tl;博士
confusionMatrix
使用預測的類別,因此有 50% 的概率截止最大限度