r-caret 中的随机搜索

Question

我最近在插入符号的 trainControl() Funktion 中发现了随机搜索选项。插入符如何生成参数以及是否有办法提供某种特定于用户的输入(例如从中采样参数的分布)？关于website我只找到这句话:

built-in models contained in caret contain code to generate random tuning parameter combinations

例如，我尝试了使用插入符和随机搜索的 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]？

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