Package | Description |
---|---|
org.apache.ignite.ml.tree |
Root package for decision trees.
|
org.apache.ignite.ml.tree.impurity.gini |
Contains Gini impurity measure and calculator.
|
org.apache.ignite.ml.tree.impurity.mse |
Contains mean squared error impurity measure and calculator.
|
Modifier and Type | Method and Description |
---|---|
protected abstract ImpurityMeasureCalculator<T> |
DecisionTree.getImpurityMeasureCalculator(Dataset<EmptyContext,DecisionTreeData> dataset)
Returns impurity measure calculator.
|
protected ImpurityMeasureCalculator<GiniImpurityMeasure> |
DecisionTreeClassificationTrainer.getImpurityMeasureCalculator(Dataset<EmptyContext,DecisionTreeData> dataset)
Returns impurity measure calculator.
|
protected ImpurityMeasureCalculator<MSEImpurityMeasure> |
DecisionTreeRegressionTrainer.getImpurityMeasureCalculator(Dataset<EmptyContext,DecisionTreeData> dataset)
Returns impurity measure calculator.
|
Modifier and Type | Class and Description |
---|---|
class |
GiniImpurityMeasureCalculator
Gini impurity measure calculator.
|
Modifier and Type | Class and Description |
---|---|
class |
MSEImpurityMeasureCalculator
Meas squared error (variance) impurity measure calculator.
|
GridGain In-Memory Computing Platform : ver. 8.9.14 Release Date : November 5 2024