public class Gini extends java.lang.Object implements IBestSplitMeasure
Constructor and Description |
---|
Gini() |
Modifier and Type | Method and Description |
---|---|
static double |
getGiniImpurity(double[] totalCounts,
double[] positiveCounts,
int kBEPPConstant,
boolean unbiasedEstimate)
Returns Gini impurity score for the N groups after a nominal split (the closer to zero, the better the split).
|
static double |
getGiniImpurity(SufficientStatistics ss,
int kBEPPConstant,
boolean unbiasedEstimate)
Returns Gini impurity score of the two groups after the split - the closer to zero, the better the split.
|
double |
getScore(double[] totalCounts,
double[] positiveCounts,
int kBEPPConstant,
boolean unbiasedEstimate)
Returns purity score for the N groups after a nominal split - larger is better
|
double |
getScore(SufficientStatistics ss,
int kBEPPConstant,
boolean unbiasedEstimate)
Returns purity score of the two groups after the split - larger is better
|
public static double getGiniImpurity(SufficientStatistics ss, int kBEPPConstant, boolean unbiasedEstimate)
public static double getGiniImpurity(double[] totalCounts, double[] positiveCounts, int kBEPPConstant, boolean unbiasedEstimate)
public double getScore(SufficientStatistics ss, int kBEPPConstant, boolean unbiasedEstimate)
getScore
in interface IBestSplitMeasure
public double getScore(double[] totalCounts, double[] positiveCounts, int kBEPPConstant, boolean unbiasedEstimate)
getScore
in interface IBestSplitMeasure