extract_params.randomForest {aurelius}R Documentation

extract_params.randomForest

Description

Extracts parameters from a forest made by the randomForest function

Usage

## S3 method for class 'randomForest'
extract_params(object, which_tree = 1, ...)

Arguments

object

an object of class "randomForest"

which_tree

the number of the tree to extract

...

further arguments passed to or from other methods

Value

a list that is extracted from the randomForest object

Examples

dat <- data.frame(X1 = runif(100), 
                  X2 = rnorm(100))
dat$Y <- factor((rexp(100,5) + 5 * dat$X1 - 4 * dat$X2) > 0)

model <- randomForest::randomForest(Y ~ X1 + X2, data=dat, ntree=10)
my_tree <- extract_params(model, 1)

[Package aurelius version 0.8.4 Index]