Ошибка: сетка параметров настройки должна иметь столбцы mtry, SVM Regression

Question

Я пытаюсь настроить регрессионную модель SVM с помощью пакета каретки. Ниже код:

control <- trainControl(method="cv", number=5)
tunegrid <- expand.grid(.mtry=c(6:12), .ntree=c(500, 600, 700, 800, 900, 1000))
set.seed(2)
custom <- train(CRTOT_03~., data=train, method="rf", metric="rmse", tuneGrid=tunegrid, trControl=control)
summary(custom)
plot(custom)

и я получаю ошибку

Error : The tuning parameter grid should have columns mtry

Answer 1

Вы используете случайные леса, а не машины опорных векторов. Вы получаете ошибку, потому что вы можете установить .mtry только в сетке настройки для случайных лесов в caret Параметр ntree устанавливается путем передачи ntree в train, например,

control <- trainControl(method="cv", number=5)
tunegrid <- expand.grid(.mtry = 6:12)
set.seed(2)
custom <- train(CRTOT_03~., 
          data=train, method="rf", 
          metric="rmse", 
          tuneGrid=tunegrid, 
          ntree = 1000,
          trControl=control)

ntree передается непосредственно randomForest