CaseBasedReasoning (version 0.1)

terminalNodeIDs: Get the terminal node id of a RandomForest Object

Description

Extracts for each observation and for each tree in the forest the terminal node id.

Usage

terminalNodeIDs(x, rfObject)

Arguments

x

a data.frame

rfObject

ranger object

Value

Matrix with terminal node IDs for all observations in x (rows) and trees (columns)

Examples

Run this code
# NOT RUN {
library(ranger)
rf.fit <- ranger(Species ~ ., data = iris, num.trees = 5, write.forest = TRUE)
dfNodes <- terminalNodeIDs(iris[, -5], rf.fit)

# }

Run the code above in your browser using DataLab