如何在r中找到预测截止点

Question

我正在训练一个glm模型，其中我的类属性'adverse_effects'是一个包含0和1的因子

Row[0].Cell[0] = "test"  Row [0].Cell[1] = "test"
Row[1].Cell[0] = "test"  Row [1].Cell[1] = "test"
Row[2].Cell[0] = "test"  Row [2].Cell[1] = need to display an image here
Row[3].Cell[0] = "test"  Row [3].Cell[1] = "test"

这是预测摘要

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])

我如何知道预测的截止点？如何将预测结果映射回0和1？

P.S我得到了一个0.6144的金额

Answer 1

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

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

自然截止值为0.5，这将选择最可能的结果。除非你有特殊的情况（比如假阴性比假阳性差），否则我会坚持使用0.5作为截止值。