Package | Description |
---|---|
org.apache.ignite.ml.tree |
Root package for decision trees.
|
org.apache.ignite.ml.tree.data |
Contains data and data builder required for decision tree trainers built on top of partition based dataset.
|
org.apache.ignite.ml.tree.impurity |
Root package for decision tree impurity measures and calculators.
|
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.
|
org.apache.ignite.ml.tree.leaf |
Root package for decision trees leaf builders.
|
Modifier and Type | Method and Description |
---|---|
default TreeFilter |
TreeFilter.and(TreeFilter other)
Returns a composed predicate.
|
Modifier and Type | Method and Description |
---|---|
default TreeFilter |
TreeFilter.and(TreeFilter other)
Returns a composed predicate.
|
Modifier and Type | Method and Description |
---|---|
TreeDataIndex |
DecisionTreeData.createIndexByFilter(int depth,
TreeFilter filter)
Builds index in according to current tree depth and cached indexes in upper levels.
|
TreeDataIndex |
TreeDataIndex.filter(TreeFilter filter)
Creates projection of current index in according to
TreeFilter . |
DecisionTreeData |
DecisionTreeData.filter(TreeFilter filter)
Filters objects and returns only data that passed filter.
|
Modifier and Type | Method and Description |
---|---|
abstract StepFunction<T>[] |
ImpurityMeasureCalculator.calculate(DecisionTreeData data,
TreeFilter filter,
int depth)
Calculates all impurity measures required required to find a best split and returns them as an array of
StepFunction (for every column). |
Modifier and Type | Method and Description |
---|---|
StepFunction<GiniImpurityMeasure>[] |
GiniImpurityMeasureCalculator.calculate(DecisionTreeData data,
TreeFilter filter,
int depth)
Calculates all impurity measures required required to find a best split and returns them as an array of
StepFunction (for every column). |
Modifier and Type | Method and Description |
---|---|
StepFunction<MSEImpurityMeasure>[] |
MSEImpurityMeasureCalculator.calculate(DecisionTreeData data,
TreeFilter filter,
int depth)
Calculates all impurity measures required required to find a best split and returns them as an array of
StepFunction (for every column). |
Modifier and Type | Method and Description |
---|---|
DecisionTreeLeafNode |
DecisionTreeLeafBuilder.createLeafNode(Dataset<EmptyContext,DecisionTreeData> dataset,
TreeFilter pred)
Creates new leaf node for given dataset and node predicate.
|
DecisionTreeLeafNode |
MeanDecisionTreeLeafBuilder.createLeafNode(Dataset<EmptyContext,DecisionTreeData> dataset,
TreeFilter pred)
Creates new leaf node for given dataset and node predicate.
|
DecisionTreeLeafNode |
MostCommonDecisionTreeLeafBuilder.createLeafNode(Dataset<EmptyContext,DecisionTreeData> dataset,
TreeFilter pred)
Creates new leaf node for given dataset and node predicate.
|
GridGain In-Memory Computing Platform : ver. 8.9.4 Release Date : April 16 2024