我最近在插入符号trainControl()
Funktion中遇到了随机搜索选项。插入符如何生成参数,是否有提供某种用户特定输入的方法(例如,从中采样参数的分布)?在website我只发现了这句话:
插入符号中包含的内置模型包含用于生成随机调整参数组合的代码
例如,我尝试使用插入符号和随机搜索mxnet
:
# Train control with random search
rs_control <- trainControl(method = "boot",
number = 2,
search = "random",
verboseIter = TRUE
)
# Training
model_fit <- train(form = y ~ .,
data = df_train,
method = "avMxnet",
preProcess = c("center", "scale"),
tuneLength = 20,
trControl = rs_control
)
使用此代码,插入符号为第一层上的神经元数量和其他参数(学习率,动量,丢失和重复)采样合理的值,但保持第二/第三层恒定为零。有没有办法告诉插入符号对所有三个层的样本进行均匀分布。 [25,150]?
答案 0 :(得分:1)
首先,由于该模型不可用,我不确定您使用的caret
版本
> library(caret)
> getModelInfo("avMxnet")
Error in getModelInfo("avMxnet") :
That model is not in caret's built-in library
这是CRAN版本caret_6.0-70。
我假设您正在使用github上的代码。如果是这种情况,您可以更改grid
代码并让它生成您喜欢的任何随机搜索网格。