r - Caret Model 如何使用优化的参数训练最终模型？

Question

train_control <- trainControl(method='cv', number=10)    
model <- train(Class ~ Age+BMI+DBP+DPF+NumPregnancies+PG2+SI2+TSFT, method ='rf',data=input,trControl=train_control)

pmml(model$finalModel)
#Error in names(field$class) <- var.names : attempt to set an attribute on NULL

当我尝试将插入符号训练的 RF 模型导出到 PMML 时，它失败了。有没有一种方法可以使用 caret 包调整的最佳参数手动运行 RF，以便我可以将模型导出到 PMML？

Answer 1

调整参数可以通过model$bestTune

> model$bestTune
mtry
3   23

Answer 2

caret 中的“rf”方法利用了 randomForest 包和 randomForest 函数。这是如果您想“手动”运行基本随机森林，而不需要插入符号。