如何在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？

PS 我的 auc 是 0.6144

Answer 1

predict.glm 采用type 参数，该参数确定进行预测的规模。我想你想要type="response"，所以试试

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

自然截止值为 0.5，这将选择最可能的结果。除非您有特殊情况（例如误报比误报更糟糕），否则我会坚持使用 0.5 作为截止值。