caret (version 5.13-037)

rfeControl: Controlling the Feature Selection Algorithms

Description

This function generates a control object that can be used to specify the details of the feature selection algorithms used in this package.

Usage

rfeControl(functions = NULL,
           rerank = FALSE,
           method = "boot",
           saveDetails = FALSE,
           number = ifelse(method %in% c("cv", "repeatedcv"), 10, 25),
           repeats = ifelse(method %in% c("cv", "repeatedcv"), 1, number),
           verbose = FALSE,
           returnResamp = "all",
           p = .75,
           index = NULL,
           timingSamps = 0)

Arguments

functions
a list of functions for model fitting, prediction and variable importance (see Details below)
rerank
a logical: should variable importance be re-calculated each time features are removed?
method
The external resampling method: boot, cv, LOOCV or LGOCV (for repeated training/test splits
number
Either the number of folds or number of resampling iterations
repeats
For repeated k-fold cross-validation only: the number of complete sets of folds to compute
saveDetails
a logical to save the predictions and variable importances from the selection process
verbose
a logical to print a log for each external resampling iteration
returnResamp
A character string indicating how much of the resampled summary metrics should be saved. Values can be ``final'', ``all'' or ``none''
p
For leave-group out cross-validation: the training percentage
index
a list with elements for each external resampling iteration. Each list element is the sample rows used for training at that iteration.
timingSamps
the number of training set samples that will be used to measure the time for predicting samples (zero indicates that the prediction time should not be estimated).

Value

  • A list

itemize

  • y

item

  • x
  • x
  • y
  • metric
  • maximize
  • size

pkg

caret

Details

Backwards selection requires function to be specified for some operations.

The fit function builds the model based on the current data set. The arguments for the function must be:

  • x
{ the current training set of predictor data with the appropriate subset of variables} y{ the current outcome data (either a numeric or factor vector)} first{ a single logical value for whether the current predictor set has all possible variables} last{ similar to first, but TRUE when the last model is fit with the final subset size and predictors.} ...{optional arguments to pass to the fit function in the call to rfe}

See Also

rfe, lmFuncs, rfFuncs, treebagFuncs, nbFuncs, pickSizeBest, pickSizeTolerance