predicted_tree_distance: Pranger
In imbs-hl/pranger: Proximities with ranger

View source: R/04_predicted_tree_distance.R

predicted_tree_distance

R Documentation

Pranger

Description

Uses predicted terminal nodes of a tree to compute the dissimilarity matrix

Usage

predicted_tree_distance(tree_index, forest, predictions, init_dist = 0)

Arguments

`tree_index`	[`integer`] A ranger tree
`forest`	[`ranger`] A ranger object
`predictions`	[`ranger`] Predictions object from class ranger
`init_dist`	[`integer`] Initial distance between in-of-bag nodes

Value

[matrix] A matrix of dissimilarities

Author(s)

Cesaire J. K. Fouodo

Examples

set.seed(1234)
train.idx <- sample(nrow(iris), 2/3 * nrow(iris))
iris.train <- iris[train.idx, ]
iris.test <- iris[-train.idx, ]
rg.iris <- ranger(Species ~ ., data = iris.train)
pred.iris <- predict(rg.iris, data = iris.test, type = "terminalNodes")
 iris_dist <- predicted_tree_distance(
   tree_index = 1,
   forest = rg.iris,
   predictions = pred.iris)

imbs-hl/pranger documentation built on May 15, 2022, 5:27 p.m.