我找不到用于调参搜索空间的机器学习模型的默认参数列表。我知道可以通过在GitHub页面上手动搜索来获取网格值,但我认为这不是正确的方法。我需要这些参数,因为我正在尝试对所有分类模型的默认配置进行执行时间估计。我在caret文档中找不到这些参数。有没有办法获取这些参数?
回答:
modelLookup()
是你需要的。这将返回每个模型可以调节的参数。
models <- modelLookup()head(models) model parameter label forReg forClass probModel1 ada iter #Trees FALSE TRUE TRUE2 ada maxdepth Max Tree Depth FALSE TRUE TRUE3 ada nu Learning Rate FALSE TRUE TRUE4 AdaBag mfinal #Trees FALSE TRUE TRUE5 AdaBag maxdepth Max Tree Depth FALSE TRUE TRUE9 adaboost nIter #Trees FALSE TRUE TRUE
通过筛选 forClass 来获取仅用于分类的模型。