【发布时间】:2018-07-05 23:25:54
【问题描述】:
我找不到用于调整搜索空间的默认机器学习模型参数列表。我知道可以检索在github page 上手动搜索的网格值,但我认为这不是正确的方法。我需要这些参数,因为我正在尝试对默认配置上的所有分类模型进行执行时间估计。我无法在caret documentation 上找到它们。有没有办法找回它们?
【问题讨论】:
标签: r machine-learning parameters r-caret
我找不到用于调整搜索空间的默认机器学习模型参数列表。我知道可以检索在github page 上手动搜索的网格值,但我认为这不是正确的方法。我需要这些参数,因为我正在尝试对默认配置上的所有分类模型进行执行时间估计。我无法在caret documentation 上找到它们。有没有办法找回它们?
【问题讨论】:
标签: r machine-learning parameters r-caret
modelLookup() 是您所需要的。这将为每个模型返回您可以调整的参数。
models <- modelLookup()
head(models)
model parameter label forReg forClass probModel
1 ada iter #Trees FALSE TRUE TRUE
2 ada maxdepth Max Tree Depth FALSE TRUE TRUE
3 ada nu Learning Rate FALSE TRUE TRUE
4 AdaBag mfinal #Trees FALSE TRUE TRUE
5 AdaBag maxdepth Max Tree Depth FALSE TRUE TRUE
9 adaboost nIter #Trees FALSE TRUE TRUE
过滤 forClass 以获取仅用于分类的模型。
【讨论】: