R 试图让插入符/rfe 工作

Question

我有一个数据集，我正在尝试使用 R 中 caret 包中的 rfe()。

x 是我尝试预测的价格。

y 是我用来进行预测的变量。

我无法让 rfe 停止给出以下错误消息:

> lmProfile2 <- rfe(x1,y1,
+ sizes = subsets,
+ rfeControl = ctrl)
Error in rfe.default(x1, y1, sizes = subsets, rfeControl = ctrl) : 
  there should be the same number of samples in x and y

这里有一些信息:

> class(x1)
[1] "data.frame"
> class(y1)
[1] "data.frame"
> nrow(x1)
[1] 500
> nrow(y1)
[1] 500
> ncol(x1)
[1] 68
> ncol(y1)
[1] 1

另外:

> y1 <- data.frame(y = tiny4[,2])
> x1 <- data.frame(tiny4[,-c(1,2)])
> subsets <- c(5,10)
> 
> ctrl <- rfeControl(functions = lmFuncs,
+ method = "cv",
+ verbose = FALSE,
+ returnResamp = "final")
> 

知道为什么我会收到这条消息吗？

Answer 1

y 应该是数字或因子向量。在这里您可以将其作为数据框。比较:

> rfe(data.frame(matrix(rnorm(100*3), ncol=3)), sample(2, 100, replace=T), sizes=1:3, rfeControl=rfeControl(functions=lmFuncs))

Recursive feature selection

Outer resampling method: Bootstrap (25 reps) 

Resampling performance over subset size:

 Variables   RMSE Rsquared  RMSESD RsquaredSD Selected
         1 0.5154  0.02120 0.02421    0.02752        *
         2 0.5162  0.02295 0.02722    0.03204         
         3 0.5162  0.02295 0.02722    0.03204         

The top 1 variables (out of 1):
   X3

对比

> rfe(data.frame(matrix(rnorm(100*3), ncol=3)), data.frame(sample(2, 100, replace=T)), sizes=1:3, rfeControl=rfeControl(functions=lmFuncs))
Error in rfe.default(data.frame(matrix(rnorm(100 * 3), ncol = 3)), data.frame(sample(2,  : 
  there should be the same number of samples in x and y