createGrid {caret}R Documentation

Tuning Parameter Grid

Description

This function creates a data frame that contains a grid of complexity parameters specific methods.

Usage

  createGrid(method, len = 3, data = NULL)

Arguments

method a string specifying which classification model to use. Possible values are: lm, rda, lda, gbm, rf, nnet, multinom, gpls, lvq, rpart, knn, pls, pam, nb, earth, treebag, svmpoly, svmradial, fda, bagEarth, bagFDA, glmboost, gamboost, blackboost, ada, ctree and cforest
len an integer specifying the number of points on the grid for each tuning parameter.
data the training data (only needed in the case where the method is cforest, earth, bagEarth, fda, bagFDA, rpart, rf or pls.

Details

A grid is created with rows corresponding to complexity parameter combinations. If the model does not use tuning parameters (like a linear model), values of NA are returned. Columns are named the same as the parameter name, but preceded by a period.

For some models, the data should be passed to the function via the data argument. In these cases, the outcome should be included in a column named .outcome.

Value

A data frame where the rows are combinations of tuning parameters and columns correspond to the parameters. The column names should be the parameter names preceded by a dot (e.g. .mtry)

Author(s)

Max Kuhn

See Also

train

Examples

createGrid("rda", 4)
createGrid("lm")
createGrid("nnet")

[Package caret version 4.10 Index]