随机搜索r-caret

时间:2016-08-04 07:47:06

标签: r machine-learning r-caret hyperparameters mxnet

我最近在插入符号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]?

1 个答案:

答案 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代码并让它生成您喜欢的任何随机搜索网格。