tree.interpreter (version 0.1.1)

trainsetBiasTree: Trainset Bias

Description

For a tree in the forest, trainset bias is the prediction of its root node, or the unconditional prediction of the tree, or the average response of the samples used to train the tree.

Usage

trainsetBiasTree(tidy.RF, tree)
trainsetBias(tidy.RF)

Arguments

tidy.RF

A tidy random forest. The random forest to extract train set bias from.

tree

An integer. The index of the tree to look at.

Value

A matrix. The content depends the type of the response.

• Regression: A 1-by-1 matrix. The trainset bias for the prediction of the response.

• Classification: A 1-by-D matrix, where D is the number of response classes. Each column of the matrix stands for the trainset bias for the prediction of each response class.

Functions

• trainsetBiasTree: Trainset bias within a single tree

• trainsetBias: Trainset bias within the whole forest

Details

For a forest, the trainset bias is simply the average trainset bias across all trees. This is because the prediction of a forest is the average of the predictions of its trees.

Together with featureContrib(Tree), they can decompose the prediction by feature importance:

$$prediction(MODEL, X) = trainsetBias(MODEL) + featureContrib_1(MODEL, X) + ... + featureContrib_p(MODEL, X),$$

where MODEL can be either a tree or a forest.

References

Interpreting random forests http://blog.datadive.net/interpreting-random-forests/

Random forest interpretation with scikit-learn http://blog.datadive.net/random-forest-interpretation-with-scikit-learn/

See Also

featureContrib

Examples

# NOT RUN {
library(ranger)
rfobj <- ranger(Species ~ ., iris, keep.inbag=TRUE)
tidy.RF <- tidyRF(rfobj, iris[, -5], iris[, 5])
trainsetBiasTree(tidy.RF, 1)
trainsetBias(tidy.RF)

# }