caretEnsemble (version 1.0.0)

caretList: Create a list of several train models from the caret package

Description

Build a list of train objects suitable for ensembling using the caretEnsemble function.

Usage

caretList(..., trControl = trainControl(), methodList = NULL,
  tuneList = NULL)

Arguments

...

arguments to pass to train. These arguments will determine which train method gets dispatched.

trControl

a trainControl object. We are going to intercept this object check that it has the "index" slot defined, and define the indexes if they are not.

methodList

optional, a character vector of caret models to ensemble. One of methodList or tuneList must be specified.

tuneList

optional, a NAMED list of caretModelSpec objects. This much more flexible than methodList and allows the specificaiton of model-specific parameters (e.g. passing trace=FALSE to nnet)

Value

• A list of train objects

Examples

myControl <- trainControl(method='cv', number=5)
caretList(
  Sepal.Length ~ Sepal.Width,
  head(iris, 50),
  methodList=c('glm', 'lm'),
  trControl=myControl
  )
caretList(
  Sepal.Length ~ Sepal.Width,
  head(iris, 50), methodList=c('lm'),
  tuneList=list(
    nnet=caretModelSpec(method='nnet', trace=FALSE, tuneLength=1)
 ),
  trControl=myControl
  )