如何找到預測截止點r中

Question

我訓練GLM模型在我的類屬性「adverse_effects」是含0和1

ctrl <- trainControl(method = "cv", number = 5) 

model_logreg <- train(adverse_effects ~.,family=binomial(link='logit'),data=trainSplit_logreg, method = "glm", trControl = ctrl) 


predictors <- names(trainSplit_logreg)[names(trainSplit_logreg) != 'adverse_effects'] 
pred_logreg <- predict(model_logreg$finalModel, testSplit_logreg[,predictors]) 

這一個因素是預測

summary(pred_logreg) 
    Min. 1st Qu. Median  Mean 3rd Qu.  Max. 
-14.5600 -2.1220 -1.8700 -1.9890 -1.7090 -0.9459 

摘要

我如何知道預測的切點？如何將預測結果映射回0和1？

P.S我得到的0.6144

Answer 1

predict.glm的AUC需要type參數，它決定了其做出預測的規模。我想你想type="response"，所以儘量

predict(model_logreg$finalModel, testSplit_logreg[,predictors], type="response") 

自然邊界值是0.5，這將選擇最有可能的結果。除非你有特殊情況（例如，假陰性比假陽性更差），否則我會堅持0.5作爲臨界值。