org.apache.spark.mllib.tree.impurity

Class VarianceCalculator

Object

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

org.apache.spark.mllib.tree.impurity.VarianceCalculator

public class VarianceCalculator
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
VarianceCalculator(double[] stats)

Method Summary

Methods

Modifier and Type	Method and Description
double	calculate() Calculate the impurity from the stored sufficient statistics.
VarianceCalculator	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.
String	toString()

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

add, prob, stats, subtract

Methods inherited from class Object

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

Constructor Detail

VarianceCalculator

public VarianceCalculator(double[] stats)

Method Detail

copy

public VarianceCalculator 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

toString

public String toString()

Overrides:

toString in class Object