org.apache.ignite.ml.tree.impurity

Interface ImpurityMeasureCalculator<T extends ImpurityMeasure<T>>

Type Parameters:

T - Type of impurity measure.

All Superinterfaces:

Serializable

All Known Implementing Classes:

GiniImpurityMeasureCalculator, MSEImpurityMeasureCalculator

public interface ImpurityMeasureCalculator<T extends ImpurityMeasure<T>>
extends Serializable

Base interface for impurity measure calculators that calculates all impurity measures required to find a best split.

Method Summary

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

Method Detail

calculate

StepFunction<T>[] 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).

Parameters:

data - Features and labels.

Returns:

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