r - 插入符 - 调整 : Models' default parameters

Question

我找不到调整搜索空间的默认机器学习模型参数列表。我知道可以通过在 github page 上手动搜索来检索网格值。 ，但我认为这不是正确的方法。我需要这些参数，因为我正在尝试对默认配置下的所有分类模型进行执行时间估计。我无法在 caret documentation 上找到它们。有办法找回吗？

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 进行过滤以获取仅用于分类的模型。