简体   繁体   中英

Caret - tuning: Models' default parameters

I can't find a list of the default machine learning models' parameters for the tuning search space. I know that it's possible to retrieve the grid values searching manually them on the github page , but I don't think this is the right way. I need these parameters because I'm trying to do an execution time estimate for all the classification models on the default configuration. I'm unable to find them on the caret documentation . Is there a way to retrieve them?

modelLookup() is what you need. This will return per model the parameters you can tune.

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

filter on forClass to get the models for classification only.

The technical post webpages of this site follow the CC BY-SA 4.0 protocol. If you need to reprint, please indicate the site URL or the original address.Any question please contact:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM