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,
           metric = NULL, 
           rerank = FALSE,
           method = "boot",
           saveDetails = FALSE,
           number = ifelse(method == "cv", 10, 25),
           verbose = TRUE,
           returnResamp = "all",
           p = .75,
           index = NULL)

Arguments

functions

a list of functions for model fitting, prediction and variable importance (see Details below)

metric

a string that specifies what summary metric will be used to select the optimal model. Possible values are "RMSE" and "Rsquared" for regression and "Accuracy" and "Kappa" for classification.

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

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''

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.

Value

A list

code

nbFuncs

itemize

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:

{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}