r - 从 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

任何帮助将不胜感激。这里是菜鸟。

谢谢。

Answer 1

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

（glmnet同时为该正则化参数的多个值拟合模型。）

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