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.