shi_ranger_forest: Pranger

View source: R/08_shi_ranger_forest.R

shi_ranger_forestR Documentation

Pranger

Description

Computes dissimilarities between individuals based on the approach of Shi and Hovarth (2006) over a forest. For each tree, dissimilarity between two individuals is increased to one if they don't belong to the same terminal nodes. Dissimilarities over the forest is obtained by summing dissimilarities obtained from each tree.

Usage

shi_ranger_forest(predictions, init_dist = 0)

Arguments

predictions

[predict.ranger] Terminal nodes predicted by predict.ranger

init_dist

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

Value

[matrix] Dissimilarity matrix according to Shi and Hovarth (2006)

Examples

library(ranger)
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, num.trees = 5)
pred.iris <- predict(rg.iris, data = iris.test, type = "terminalNodes")
shi_dist <- shi_ranger_forest(predictions = pred.iris)

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