从R glmnet对象获取预测类

Question

我试图在R中使用glmnet构建简单的多类逻辑回归模型.但是当我尝试预测测试数据并获得列联表时,我得到一个错误.下面再现了一个示例会话.

> mat = matrix(1:100,nrow=10)
> test = matrix(1:50,nrow=5)

> classes <- as.factor(11:20)

> model <- glmnet(mat, classes, family="multinomial", alpha=1)
> pred <- predict(model, test)
> table(pred, as.factor(11:15))
  Error in table(pred, as.factor(11:15)) : 
  all arguments must have the same length

任何帮助将不胜感激.R noob在这里.

谢谢.

Answer 1

对象的predict方法glmnet要求您为参数指定一个值s,该值指示要为其预测的正则化参数的哪些值.

(glmnet同时为该正则化参数的几个值拟合模型.)

因此,如果未指定值s,则predict.glmnet返回所有值的预测.如果您只需要一组预测,则需要为s调用时设置一个值predict,或者需要在事后提取相关列.