org.apache.ignite.ml.tree.impurity.mse

Class MSEImpurityMeasureCalculator

java.lang.Object

org.apache.ignite.ml.tree.impurity.mse.MSEImpurityMeasureCalculator

All Implemented Interfaces:

Serializable, ImpurityMeasureCalculator<MSEImpurityMeasure>

public class MSEImpurityMeasureCalculator
extends Object
implements ImpurityMeasureCalculator<MSEImpurityMeasure>

Meas squared error (variance) impurity measure calculator.

See Also:

Serialized Form

Constructor Summary

Constructors

Constructor and Description
MSEImpurityMeasureCalculator()

Method Summary

Modifier and Type	Method and Description
StepFunction<MSEImpurityMeasure>[]	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

MSEImpurityMeasureCalculator

public MSEImpurityMeasureCalculator()

Method Detail

calculate

public StepFunction<MSEImpurityMeasure>[] 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<MSEImpurityMeasure>

Parameters:

data - Features and labels.

Returns:

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