specmine (version 4.0.0)

feature_selection: Auto-exported function: feature_selection

Description

Auto-exported function: feature_selection

Usage

feature_selection(
  dataset,
  column.class,
  method = "rfe",
  functions,
  validation = "cv",
  repeats = 5,
  number = 10,
  subsets = 2^(2:4)
)

Value

A feature selection object produced by caret. For method = "rfe", this is an rfe object; for method = "filter", this is an sbf object.

Examples

datamat <- matrix(
  rnorm(20),
  nrow = 4,
  dimnames = list(paste0("x", 1:4), paste0("s", 1:5))
)
metadata <- data.frame(class = factor(c("A", "A", "B", "B", "A")))
dataset <- list(data = datamat, metadata = metadata)
feature_selection(dataset, "class", method = "filter", functions = caret::rfSBF)