插入調整：模型的默認參數

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以獲取僅用於分類的模型。