2016-08-04 78 views
1

我最近遇到了插入符號trainControl()函數中的隨機搜索選項。插入符號如何生成參數,以及是否有方法提供某種用戶特定的輸入(例如,參數從哪個分佈中採樣)?在website我才發現這句話:在r-caret中隨機搜索

包含在插入符號

內置機型中包含的代碼來生成隨機調整參數組合

例如,我試圖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 
        )  

使用此代碼,插入符號採集第一層神經元數量和其他參數(學習速率,動量,丟失和重複)的合理值,但保留第二個/ th第三層常數爲零。有沒有一種方法可以告訴脫字符號從所有的三層均勻分佈的樣本中進行取樣。 [25,150]?

回答

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的代碼,讓它生成任意你喜歡的隨機搜索網格。

+0

我正在使用CRAN版本caret_6.0-68和'getModelInfo(「avMxnet」)'ist實際上爲我工作..也許你必須安裝mxnet在第一個地方...你是什麼意思與代碼在github上? – winwin

+0

我很想擁有mxnet的光標! ;) – Richard

+0

據我所說,它已經被支持:http://topepo.github.io/caret/modelList.html – winwin