R神经网络插入符号参数调整错误

Question

我正在为波士顿数据集上的神经网络练习调整参数。我一直收到一个持续错误：

Error: The tuning parameter grid should have columns size, decay

以下是我的 Caret 调整设置：

caret_control <- trainControl(method = "repeatedcv",
                       number = 10,
                       repeats = 3)

caret_grid <- expand.grid(batch_size=seq(60,120,20),
                      dropout=0.5,
                      size=100,
                      decay = 0,
                      lr=2e-6,
                      activation = "relu")

caret_t <- train(medv ~ ., data = chasRad, 
             method = "nnet", 
             metric="RMSE",
             trControl = caret_control, 
             tuneGrid = caret_grid,
             verbose = FALSE)

这里的chasRad是一个12x506的矩阵。谁能帮助修复似乎由扩展网格触发的错误？

Answer 1

您收到的错误应解释为：

"The tuning parameter grid should ONLY have columns size, decay"。

您正在传递 nnet 无法在 caret 中调整的四个额外参数。有关可调参数的完整列表，运行 modelLookup(model = 'nnet').

要仅调整 size 和 decay，请将您的 caret_grid 替换为：

caret_grid <- expand.grid(size=seq(from = 1, to = 10, by = 1),
                      decay = seq(from = 0.1, to = 0.5, by = 0.1))

您的代码将运行。

R神经网络插入符号参数调整错误

R Error with neural network caret parameter tuning

r

neural-network

r-caret