我正在为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矩阵。任何人都可以帮助修复看似由扩展网格触发的错误吗?
答案 0 :(得分:1)
您获得的错误应解释为:
“调整参数网格 ONLY 应该具有列大小,衰减”。
您传递了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))
您的代码将会运行。