【发布时间】:2021-03-23 06:02:31
【问题描述】:
我有以下具有默认参数的随机森林(回归)模型
set.seed(42)
# Define train control
trControl <- trainControl(method = "cv",
number = 10,
search = "grid")
# Random Forest (regression) model
rf_reg <- train(Price.Gas~., data=data_train,
method = "rf",
metric = "RMSE",
trControl = trControl)
我希望模型通过更改其调整参数(例如 ntree、maxnodes、搜索等)来表现更好。
我不认为一一改变是最有效的方法。
如何有效地测试 R 中的参数以获得更好的随机森林(即能够很好地预测数据的随机森林)?
【问题讨论】:
标签: r machine-learning random-forest decision-tree grid-search