Я сделал пример, используя iris
и следуя инструкции . Возможно, ваша ошибка в:
dtrain [, fields, with = F]
См. Пример ниже, используя iris
:
set.seed(1)
library(caret)
nfields <- 5
control <- rfeControl(functions = rfFuncs,
method = "repeatedcv",
repeats = 1,
verbose = F)
irisx <- iris[,1:4]
fields <- colnames(irisx)[!colnames(irisx) %in% "Petal.Width"]
predictors_rfe <- rfe(irisx[,fields],
irisx$Petal.Width,
rfeControl = control)
predictors_rfe
> predictors_rfe
Recursive feature selection
Outer resampling method: Cross-Validated (10 fold, repeated 1 times)
Resampling performance over subset size:
Variables RMSE Rsquared MAE RMSESD RsquaredSD MAESD Selected
3 0.196 0.9418 0.1519 0.03502 0.0177 0.02608 *
The top 3 variables (out of 3):
Petal.Length, Sepal.Length, Sepal.Width
Если вы можете предоставить воспроизводимый пример с вашим набором данных, я буду возможность лучше проверить возможную ошибку.