org.apache.ignite.ml.tree.impurity.gini

Class GiniImpurityMeasureCalculator

java.lang.Object

org.apache.ignite.ml.tree.impurity.gini.GiniImpurityMeasureCalculator

All Implemented Interfaces:

Serializable, ImpurityMeasureCalculator<GiniImpurityMeasure>

public class GiniImpurityMeasureCalculator
extends Object
implements ImpurityMeasureCalculator<GiniImpurityMeasure>

Gini impurity measure calculator.

See Also:

Serialized Form

Constructor Summary

Constructors

Constructor and Description
GiniImpurityMeasureCalculator(Map<Double,Integer> lbEncoder) Constructs a new instance of Gini impurity measure calculator.

Method Summary

Modifier and Type	Method and Description
StepFunction<GiniImpurityMeasure>[]	calculate(DecisionTreeData data) Calculates all impurity measures required required to find a best split and returns them as an array of StepFunction (for every column).

Methods inherited from class java.lang.Object

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

Constructor Detail

GiniImpurityMeasureCalculator

public GiniImpurityMeasureCalculator(Map<Double,Integer> lbEncoder)

Constructs a new instance of Gini impurity measure calculator.

Parameters:

lbEncoder - Label encoder which defines integer value for every label class.

Method Detail

calculate

public StepFunction<GiniImpurityMeasure>[] calculate(DecisionTreeData data)

Calculates all impurity measures required required to find a best split and returns them as an array of StepFunction (for every column).

Specified by:

calculate in interface ImpurityMeasureCalculator<GiniImpurityMeasure>

Parameters:

data - Features and labels.

Returns:

Impurity measures as an array of StepFunction (for every column).