mlpack  2.0.1
Static Public Member Functions | List of all members
mlpack::tree::InformationGain 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...
 

Detailed Description

Definition at line 21 of file information_gain.hpp.

Member Function Documentation

static double mlpack::tree::InformationGain::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
countsMatrix of sufficient statistics.

Definition at line 33 of file information_gain.hpp.

static double mlpack::tree::InformationGain::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.)

Definition at line 86 of file information_gain.hpp.


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