org.apache.spark.mllib.tree.impurity

Class GiniCalculator

Object

org.apache.spark.mllib.tree.impurity.ImpurityCalculator

org.apache.spark.mllib.tree.impurity.GiniCalculator

public class GiniCalculator
extends ImpurityCalculator

Stores statistics for one (node, feature, bin) for calculating impurity. Unlike GiniAggregator, this class stores its own data and is for a specific (node, feature, bin).

Constructor Summary

Constructors

Constructor and Description
GiniCalculator(double[] stats)

Method Summary

Methods

Modifier and Type	Method and Description
double	calculate() Calculate the impurity from the stored sufficient statistics.
GiniCalculator	copy() Make a deep copy of this ImpurityCalculator.
long	count() Number of data points accounted for in the sufficient statistics.
double	predict() Prediction which should be made based on the sufficient statistics.
double	prob(double label) Probability of the label given by predict.
String	toString()

Methods inherited from class org.apache.spark.mllib.tree.impurity.ImpurityCalculator

add, stats, subtract

Methods inherited from class Object

equals, getClass, hashCode, notify, notifyAll, wait, wait, wait

Constructor Detail

GiniCalculator

public GiniCalculator(double[] stats)

Method Detail

copy

public GiniCalculator copy()

Make a deep copy of this ImpurityCalculator.

Specified by:

copy in class ImpurityCalculator

calculate

public double calculate()

Calculate the impurity from the stored sufficient statistics.

Specified by:

calculate in class ImpurityCalculator

count

public long count()

Number of data points accounted for in the sufficient statistics.

Specified by:

count in class ImpurityCalculator

predict

public double predict()

Prediction which should be made based on the sufficient statistics.

Specified by:

predict in class ImpurityCalculator

prob

public double prob(double label)

Probability of the label given by predict.

Overrides:

prob in class ImpurityCalculator

toString

public String toString()

Overrides:

toString in class Object