探索更广的模型空间

在 R 中使用 caret 的机器学习

Zach Mayer

Data Scientist at DataRobot and co-author of caret

随机森林需要调参

# Load some data
library(caret)
library(mlbench)
data(Sonar)

# Fit a model with a deeper tuning grid
model <- train(
  Class ~ ., 
  data = Sonar, 
  method = "ranger", 
  tuneLength = 10
)

# Plot the results
plot(model)

图片为折线图，展示自助法准确率随随机选取参数数量的变化。在 x≈14 达到峰值，随后迅速下降。

在 R 中使用 caret 的机器学习