Class for building model of the decision forest classification algorithm
More...
Constructs the decision forest classification model builder
- Parameters
-
context | Context to manage decision forest classification model builder |
nClasses | Number of classes |
nTrees | Number of trees |
long addLeafNode |
( |
long |
treeId, |
|
|
long |
parentId, |
|
|
long |
position, |
|
|
long |
classLabel |
|
) |
| |
Create Leaf node and add it to certain tree
- Parameters
-
treeId | Tree to which new node is added |
parentId | Parent node to which new node is added (use noParent for root node) |
position | Position in parent (e.g. 0 for left and 1 for right child in a binary tree) |
classLabel | Class label to be predicted |
- Returns
- Positive number node identifier
long addSplitNode |
( |
long |
treeId, |
|
|
long |
parentId, |
|
|
long |
position, |
|
|
long |
featureIndex, |
|
|
double |
featureValue |
|
) |
| |
Create Split node and add it to certain tree
- Parameters
-
treeId | Tree to which new node is added |
parentId | Parent node to which new node is added (use noParent for root node) |
position | Position in parent (e.g. 0 for left and 1 for right child in a binary tree) |
featureIndex | Feature index for spliting |
featureValue | Feature value for spliting |
- Returns
- Positive number node identifier
long createTree |
( |
long |
nNodes | ) |
|
Creates certain tree in the decision forest model
- Parameters
-
nNodes | Number of nodes in created tree |
- Returns
- Positive number tree identifier
Get built model of decision forest classification
- Returns
- Model of decision forest classification
The documentation for this class was generated from the following file:
- decision_forest/classification/ModelBuilder.java