predicted_forest_distance: Pranger

View source: R/05_predicted_forest_distance.R

predicted_forest_distanceR Documentation

Pranger

Description

Computes dissimilarity matrix over the forest. Note that this function will parallelize the pairwise computations of the distances between the nodes on the available CPUs.

Usage

predicted_forest_distance(predictions, forest, init_dist = 0)

Arguments

predictions

[ranger] An ranger predicted object of terminal nodes

forest

[ranger] A ranger forest

init_dist

[integer] Initial distance between in-of-bag nodes

Value

[matrix] Matrix of dissimilarities over the forest

Author(s)

Cesaire J. K. Fouodo

Examples

## Not run: 
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_forest_distance(
   forest = rg.iris,
   predictions = pred.iris)

## End(Not run)

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