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

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时，它失败了。有没有一种方法可以使用由插入符号包调整的最佳参数手动运行RF，以便我可以将模型导出到PMML？

Answer 1

可以通过model$bestTune

访问调谐参数

> model$bestTune
mtry
3   23

Answer 2

＆＃34; rf＆＃34;插入符中的方法使用randomForest包和randomForest函数。这是你想要手动运行基本随机森林＆＃34;而没有插入符号。