r - 在 R 的 GLMNET 包中预测拟合模型时出现问题

Question

我试图通过使用 R 的 glmnet 包中的岭回归，根据多个变量来预测汽车的 mpg。我已经将数据分为训练数据和测试数据，并对分类变量进行了虚拟编码。

我按照如下方式拟合交叉验证模型:

require("glmnet")

x <- as.matrix(data.frame(cylinderDummy[,2:ncol(cylinderDummy)], trainData$displacement,
            trainData$horsepower, trainData$weight, trainData$acceleration,
            originDummy[,2:ncol(originDummy)]))
y <- trainData$mpg
cv.fit <- cv.glmnet(x, y, alpha = 1, nfolds=5,type.measure="mse")

这一切都很好，但是，当我尝试对拟合模型的测试数据使用 predict() 函数时，就会出现问题:

prediction <- predict(cv.fit, testData$mpg, s="lambda.1se")

我收到以下错误:

Error in as.matrix(cbind2(1, newx) %*% nbeta) : 
error in evaluating the argument 'x' in selecting a method for function 'as.matrix': 
Error in t(.Call(Csparse_dense_crossprod, y, t(x))) : 
error in evaluating the argument 'x' in selecting a method for function 't': 
Error: Cholmod error 'X and/or Y have wrong dimensions' at file
../MatrixOps/cholmod_sdmult.c, line 90

谁能告诉我我做错了什么？谢谢!

Answer 1

prediction <- predict(cv.fit, testData$mpg, s="lambda.1se")

看来 testData$mpg 是一个向量，模型应该使用整个测试数据集来预测，而不是单个 mpg 值。

就你的情况而言，应该是这样的

testdata <- as.matrix(data.frame(cylinderDummy[,2:ncol(cylinderDummy)], testData$displacement,
        testData$horsepower, testData$weight, testData$acceleration,
        originDummy[,2:ncol(originDummy)]))
prediction <- predict(cv.fit, testData, s="lambda.1se")