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

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以获取仅用于分类的模型.