插入调整：模型的默认参数

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

过滤onClass以获取仅用于分类的模型。