如何调整火车功能的方法中的参数（插入符号）

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)