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

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

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