fuzzyforest (version 1.0.0)

screen_control: Set Parameters for Screening Step of Fuzzy Forests

Description

Creates screen_control object for controlling how feature selection will be carried out on each module.

Usage

screen_control(drop_fraction = 0.25, keep_fraction = 0.05,
  mtry_factor = 1, min_ntree = 5000, ntree_factor = 10)

Arguments

drop_fraction

A number between 0 and 1. Percentage of features dropped at each iteration.

keep_fraction

A number between 0 and 1. Proportion of features from each module that are retained from screening step.

mtry_factor

In the case of regression, mtry is set to ceiling($\sqrt(p)$*mtry_factor). In the case of classification, mtry is set to ceiling((p/3)*mtry_factor). If either of these numbers

min_ntree

Minimum number of trees grown in each random forest.

ntree_factor

A number greater than 1. ntree for each random forest is ntree_factor times the number of features. For each random forest, ntree is set to max(min_ntree, ntree_factor*p).

Value

• An object of type screen_control.

References

Daniel Conn, Tuck Ngun, Christina M. Ramirez (2015). Fuzzy Forests: a New WGCNA Based Random Forest Algorithm for Correlated, High-Dimensional Data, Journal of Statistical Software, Manuscript in progress.

Examples

drop_fraction <- .25
keep_fraction <- .1
mtry_factor <- 1
min_ntree <- 5000
ntree_factor <- 5
screen_params <- screen_control(drop_fraction=drop_fraction,
                                keep_fraction=keep_fraction,
                                mtry_factor=mtry_factor,
                                min_ntree=min_ntree,
                                ntree_factor=ntree_factor)