org.apache.spark.mllib.tree.impurity

Class GiniAggregator

Object

org.apache.spark.mllib.tree.impurity.ImpurityAggregator

org.apache.spark.mllib.tree.impurity.GiniAggregator

All Implemented Interfaces:

java.io.Serializable

public class GiniAggregator
extends ImpurityAggregator
implements scala.Serializable

Class for updating views of a vector of sufficient statistics, in order to compute impurity from a sample. Note: Instances of this class do not hold the data; they operate on views of the data.

See Also:

Serialized Form

Constructor Summary

Constructors

Constructor and Description
GiniAggregator(int numClasses)

Method Summary

Methods

Modifier and Type	Method and Description
GiniCalculator	getCalculator(double[] allStats, int offset) Get an ImpurityCalculator for a (node, feature, bin).
void	update(double[] allStats, int offset, double label, double instanceWeight) Update stats for one (node, feature, bin) with the given label.

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

merge, statsSize

Methods inherited from class Object

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

Constructor Detail

GiniAggregator

public GiniAggregator(int numClasses)

Method Detail

update

public void update(double[] allStats,
          int offset,
          double label,
          double instanceWeight)

Update stats for one (node, feature, bin) with the given label.

Specified by:

update in class ImpurityAggregator

Parameters:

allStats - Flat stats array, with stats for this (node, feature, bin) contiguous.

offset - Start index of stats for this (node, feature, bin).

getCalculator

public GiniCalculator getCalculator(double[] allStats,
                           int offset)

Get an ImpurityCalculator for a (node, feature, bin).

Specified by:

getCalculator in class ImpurityAggregator

Parameters:

allStats - Flat stats array, with stats for this (node, feature, bin) contiguous.

offset - Start index of stats for this (node, feature, bin).