Случайный поиск в r-caret

Question

Я недавно наткнулся на случайный поиск в Caret's 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]?

Answer 1

Во-первых, я не уверен, какая версия caret вы используете, так как эта модель не доступна

> library(caret) 
> getModelInfo("avMxnet") 
Error in getModelInfo("avMxnet") : 
    That model is not in caret's built-in library 

Это с версией caret_6.0-70 CRAN.

Я предполагаю, что вы используете код на github. Если это так, вы можете изменить код grid и заставить его генерировать любую случайную сетку поиска, которая вам нравится.