如何調整火車功能的方法中的參數（插入符號）

Question

我正在嘗試使用插入符號包進行交叉驗證重采樣，但是我想訓練的功能需要一些參數。 我想使用NMF軟件包中的函數nmf：

res <- nmf(dfrm, 6, nrun=20)

和代碼

inTraining <- createDataPartition(dfrm, p = .8, list = FALSE, times = 200)
training <- dfrm[ inTraining,]
testing  <- dfrm[-inTraining,]

fitControl <- trainControl(## 5-fold cv
                       method = "cv",
                       number = 5
                       )

cv_result <- train(dfrm.T, res, 
             data = training,
             method = "nmf",
             trControl = fitControl)

但是，如果僅將nmf放入方法中，則不會使用6和nrun = 20。 我該如何調整？ 謝謝

Answer 1

nmf在插入符號中不是可用的方法。

可以在火車上使用其他功能參數。 例如，帶有randomForest的ntree參數。 僅在mtry上進行調優，但是您可以添加其他功能參數。

train(Species ~ ., data = iris, method = "rf", ntree = 10)