InfGain: computes information gain for each parent node in a tree

In markusloecher/rfVarImpOOB: Unbiased Variable Importance for Random Forests

Description

information gain for each parent node in a tree

Usage

InfGain(tree, total = TRUE, zeroLeaf = TRUE, score = c("PMDI21", 


    "MDI", "MDA", "MIA")[1], verbose = 0)

Arguments

tree	tree (from randomForest::getTree)
total	if TRUE compute the sum instead of the mean
zeroLeaf	if TRUE discard the information gain due to splits resulting in n=1
score	scoring method:MDI=mean decrease impurity (Gini),MDA=mean decrease accuracy (permutation),MIA=mean increase accuracy
verbose	level of verbosity

Value

tree object augmented with information gain at each node

Author(s)

Markus Loecher <[e-mail]>

markusloecher/rfVarImpOOB documentation built on July 5, 2020, 6:50 p.m.