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... | |
Detailed Description
Definition at line 19 of file information_gain.hpp.
Member Function Documentation
◆ Evaluate()
|
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.
Definition at line 31 of file information_gain.hpp.
◆ Range()
|
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 84 of file information_gain.hpp.
The documentation for this class was generated from the following file:
- /home/ryan/src/mlpack.org/_src/mlpack-git/src/mlpack/methods/hoeffding_trees/information_gain.hpp