Package | Description |
---|---|
org.apache.ignite.ml.tree.randomforest |
Contains random forest implementation classes.
|
org.apache.ignite.ml.tree.randomforest.data |
Package contains helper data structures for random forest implementation.
|
org.apache.ignite.ml.tree.randomforest.data.impurity |
Contains implementation of impurity computers based on histograms.
|
Modifier and Type | Method and Description |
---|---|
protected ArrayList<TreeRoot> |
RandomForestTrainer.initTrees(Queue<TreeNode> treesQueue)
Creates list of trees.
|
T |
RandomForestTrainer.withNodesToLearnSelectionStrgy(Function<Queue<TreeNode>,List<TreeNode>> strgy)
Sets strategy for selection nodes from learning queue in each iteration.
|
T |
RandomForestTrainer.withNodesToLearnSelectionStrgy(Function<Queue<TreeNode>,List<TreeNode>> strgy)
Sets strategy for selection nodes from learning queue in each iteration.
|
Modifier and Type | Method and Description |
---|---|
TreeNode |
TreeNode.getLeft() |
TreeNode |
TreeNode.getRight() |
TreeNode |
TreeRoot.getRootNode() |
Modifier and Type | Method and Description |
---|---|
List<TreeNode> |
TreeRoot.getLeafs() |
List<TreeNode> |
NodeSplit.split(TreeNode node)
Split node from parameter onto two children nodes.
|
List<TreeNode> |
TreeNode.toConditional(int featureId,
double val)
Convert node to conditional node.
|
Modifier and Type | Method and Description |
---|---|
void |
NodeSplit.createLeaf(TreeNode node)
Convert node to leaf.
|
List<TreeNode> |
NodeSplit.split(TreeNode node)
Split node from parameter onto two children nodes.
|
Constructor and Description |
---|
TreeRoot(TreeNode root,
Set<Integer> usedFeatures)
Create an instance of TreeRoot.
|
Modifier and Type | Method and Description |
---|---|
Map<NodeId,ImpurityHistogramsComputer.NodeImpurityHistograms<S>> |
ImpurityHistogramsComputer.aggregateImpurityStatistics(ArrayList<TreeRoot> roots,
Map<Integer,BucketMeta> histMeta,
Map<NodeId,TreeNode> nodesToLearn,
Dataset<EmptyContext,BootstrappedDatasetPartition> dataset)
Computes histograms for each feature.
|
GridGain In-Memory Computing Platform : ver. 8.9.15 Release Date : December 3 2024