从R glmnet对象中获取预测类

Question

我正在尝试使用R中的glmnet构建简单的多类logistic回归模型。然而，当我试图预测测试数据并获得列联表时，我得到了一个错误。下面是一个示例会话。

> 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

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

(glmnet同时为此正则化参数的多个值拟合模型。)

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