使用R

Question

我在R中的cv.glmnet对象上使用预测函数 我有一个我试图预测的二元结果向量（例如0 1 1 0 1）。 预测函数是否默认为查找默认预测1或0的预测概率？我找不到能够回答问题的文档。

具有以下一般概念的代码：

for(holdout in 1:nrow(data)){
    myglmnet = cv.glmnet(data[-holdout, ], matrix(outcome[-holdout], nrow=1), family = "binomial", type.measure="mae", grouped=FALSE, alpha=1, nfolds=nrow(data))

    glmnet_hat[holdout] <- as.numeric(predict(myglmnet, matrix(data[holdout, ], nrow=1), type="response", s="lambda.min"))
}

glmnet_hat是否预测了0或1响应的概率？我可以链接到的任何文档源也都很棒。谢谢！

Answer 1

在上面的代码中，您通过指定type= "response"来建模概率结果。根据{{​​3}}，对于二元结果，您可以使用type ="response"来获取拟合概率，或使用type ="class"来获得与最大概率相对应的类标签。