public class InfoGainSplitMetric extends SplitMetric implements java.io.Serializable
Constructor and Description |
---|
InfoGainSplitMetric(double minFracWeightForTwoBranches) |
Modifier and Type | Method and Description |
---|---|
double |
evaluateSplit(java.util.Map<java.lang.String,WeightMass> preDist,
java.util.List<java.util.Map<java.lang.String,WeightMass>> postDist)
Evaluate the merit of a split
|
double |
getMetricRange(java.util.Map<java.lang.String,WeightMass> preDist)
Get the range of the splitting metric
|
sum
public InfoGainSplitMetric(double minFracWeightForTwoBranches)
public double evaluateSplit(java.util.Map<java.lang.String,WeightMass> preDist, java.util.List<java.util.Map<java.lang.String,WeightMass>> postDist)
SplitMetric
evaluateSplit
in class SplitMetric
preDist
- the class distribution before the splitpostDist
- the class distributions after the splitpublic double getMetricRange(java.util.Map<java.lang.String,WeightMass> preDist)
SplitMetric
getMetricRange
in class SplitMetric
preDist
- the pre-split class distribution