| Package | Description |
|---|---|
| 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 |
|---|---|
DecisionTreeData |
DecisionTreeDataBuilder.build(Iterator<UpstreamEntry<K,V>> upstreamData,
long upstreamDataSize,
C ctx)
Builds a new partition
data from a partition upstream data and partition context |
DecisionTreeData |
DecisionTreeData.filter(TreeFilter filter)
Filters objects and returns only data that passed filter.
|
| Modifier and Type | Method and Description |
|---|---|
StepFunction<T>[] |
ImpurityMeasureCalculator.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). |
| Modifier and Type | Method and Description |
|---|---|
StepFunction<GiniImpurityMeasure>[] |
GiniImpurityMeasureCalculator.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). |
| Modifier and Type | Method and Description |
|---|---|
StepFunction<MSEImpurityMeasure>[] |
MSEImpurityMeasureCalculator.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). |
| 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 |
MostCommonDecisionTreeLeafBuilder.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.
|
Follow @ApacheIgnite
Ignite Fabric : ver. 2.5.0 Release Date : May 23 2018