Feature Selection Filter

Feature Filters quantify the importance of each feature of a Task by assigning them a numerical score. In a second step, features can be selected by either selecting a fixed absolute or relative frequency of the best features, or by thresholding on the score value.

The Filter PipeOp allows to use filters as a preprocessing step.

Example Usage

Use the \(-\log_{10}()\)-transformed \(p\)-values of a Kruskal-Wallis rank sum test (implemented in kruskal.test()) for filtering features of the Diabetes tasks.

library("mlr3verse")

Loading required package: mlr3

# retrieve a task
task = tsk("diabetes")

# retrieve a filter
filter = flt("kruskal_test")

# calculate scores
filter$calculate(task)

# access scores
filter$scores

  glucose       age   insulin      mass  pregnant   triceps  pressure  pedigree 
37.030758 14.442510 12.035615  9.658979  5.608412  5.074205  2.445548  2.247475 

# plot scores
autoplot(filter)

# subset task to 3 most important features
task$select(head(names(filter$scores), 3))
task$feature_names

[1] "age"     "glucose" "insulin"