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)

如何调整列车功能方法中的参数（插入包）

1 个答案: