caretEnsemble (version 2.0.3)

caretList: Create a list of several train models from the caret package Build a list of train objects suitable for ensembling using the caretEnsemble function.

Description

Create a list of several train models from the caret package Build a list of train objects suitable for ensembling using the caretEnsemble function.

Usage

caretList(
  ...,
  trControl = NULL,
  methodList = NULL,
  tuneList = NULL,
  continue_on_fail = FALSE
)

Value

A list of train objects. If the model fails to build, it is dropped from the list.

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 specification of model-specific parameters (e.g. passing trace=FALSE to nnet)

continue_on_fail,

logical, should a valid caretList be returned that excludes models that fail, default is FALSE

Examples

if (FALSE) {
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
  )
  }