Caret调优:模型的默认参数

Question

我找不到用于优化搜索空间的默认机器学习模型参数的列表。我知道可以在github页面上手动检索网格值，但我认为这不是正确的方法。我需要这些参数，因为我试图对默认配置中的所有分类模型执行时间估计。我在卡雷特文件上找不到他们。有办法找回它们吗？

Answer 1

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进行过滤，得到仅供分类的模型。