caret (version 3.12)

createGrid: 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
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 rpart, rf or pls.

Value

  • A data frame

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.

See Also

train

Examples

Run this code
createGrid("rda", 4)
createGrid("lm")
createGrid("nnet")

Run the code above in your browser using DataCamp Workspace