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

时间:2018-04-03 16:21:22

标签: r neural-network r-caret

我正在为Boston数据集上的神经网络练习调整参数。我一直遇到一个持续的错误:

  

错误:调整参数网格应具有列大小,衰减

以下是我的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矩阵。任何人都可以帮助修复看似由扩展网格触发的错误吗?

1 个答案:

答案 0 :(得分:1)

您获得的错误应解释为:

“调整参数网格 ONLY 应该具有列大小,衰减”。

您传递了nnet无法调入caret的其他四个参数。有关可调参数的完整列表,请运行modelLookup(model = 'nnet')

要仅调整sizedecay,请将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))

您的代码将会运行。

相关问题