shi_ranger_one_tree: Pranger

In imbs-hl/pranger: Proximities with ranger

Pranger

Description

Given a tree, this function computes dissimilarities between individuals based on the approach of Shi and Hovarth (2006). That is, dissimilarity between two individuals is increased to one if they don't belong to the same terminal nodes

Usage

shi_ranger_one_tree(tree_index = 1, predictions, init_dist = 0)

Arguments

tree_index	[integer] Tree index for which the dissimilarity matrix is required
predictions	[predict.ranger] Terminal nodes predicted by predict.ranger
init_dist	[integer] Initial distance between in-of-bag individual. Required if oob = TRUE.

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)
pred.iris <- predict(rg.iris, data = iris.test, type = "terminalNodes")
shi_dist <- shi_ranger_one_tree(tree_index = 1,
                 predictions = pred.iris)

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