extract_params {forestControl}

Extract forest parameters

Description

For a randomForest or ranger classification object, extract the parameters needed to calculate an approximate selection frequency threshold

Usage

extract_params(x)

Arguments

x

a randomForest, ranger or parsnip object

Value

a list of four elements

• Fn The number of features considered at each internal node (mtry)

• Ft The total number of features in the data set

• K The average number of binary tests/internal nodes across the enitre forest

• Tr The total number of trees in the forest

Author(s)

Tom Wilson [e-mail]

Examples

library(randomForest)
data(iris)
iris.rf <- randomForest(iris[,-5], iris[,5], forest = TRUE)

iris.params <- extract_params(iris.rf)
print(iris.params)