mlpack::tree::HoeffdingInformationGain Class Reference

Static Public Member Functions
static double	Evaluate (const arma::Mat< size_t > &counts)
	Given the sufficient statistics of a proposed split, calculate the information gain if that split was to be used. More...
static double	Range (const size_t numClasses)
	Return the range of the information gain for the given number of classes. More...

Member Function Documentation

Evaluate()

static double mlpack::tree::HoeffdingInformationGain::Evaluate ( const arma::Mat< size_t > &  counts )

inlinestatic

Given the sufficient statistics of a proposed split, calculate the information gain if that split was to be used.

The 'counts' matrix should contain the number of points in each class in each column, so the size of 'counts' is children x classes, where 'children' is the number of child nodes in the proposed split.

Parameters

counts

Matrix of sufficient statistics.

Range()

static double mlpack::tree::HoeffdingInformationGain::Range ( const size_t  numClasses )

inlinestatic

Return the range of the information gain for the given number of classes.

(That is, the difference between the maximum possible value and the minimum possible value.)

---

The documentation for this class was generated from the following file:

• src/mlpack/methods/hoeffding_trees/information_gain.hpp