Model Builder class for Decision Forest Classification Model algorithm
More...
|
typedef size_t | NodeId |
| Node identifier type is size_t
|
|
typedef size_t | TreeId |
| Tree identifier type is size_t
|
|
Constructs the Decision forest classification model builder
- Parameters
-
[in] | nClasses | Number of classes |
[in] | nTrees | Number of trees in model |
NodeId addLeafNode |
( |
const TreeId |
treeId, |
|
|
const NodeId |
parentId, |
|
|
const size_t |
position, |
|
|
const size_t |
classLabel |
|
) |
| |
|
inline |
Create Leaf node and add it to certain tree
- Parameters
-
[in] | treeId | Tree to which new node is added |
[in] | parentId | Parent node to which new node is added (use noParent for root node) |
[in] | position | Position in parent (e.g. 0 for left and 1 for right child in a binary tree) |
[in] | classLabel | Class label to be predicted |
- Returns
- Node identifier
NodeId addLeafNodeByProba |
( |
const TreeId |
treeId, |
|
|
const NodeId |
parentId, |
|
|
const size_t |
position, |
|
|
const double *const |
proba |
|
) |
| |
|
inline |
Create Leaf node and add it to certain tree
- Parameters
-
[in] | treeId | Tree to which new node is added |
[in] | parentId | Parent node to which new node is added (use noParent for root node) |
[in] | position | Position in parent (e.g. 0 for left and 1 for right child in a binary tree) |
[in] | proba | Array with probability values for each class |
- Returns
- Node identifier
NodeId addSplitNode |
( |
const TreeId |
treeId, |
|
|
const NodeId |
parentId, |
|
|
const size_t |
position, |
|
|
const size_t |
featureIndex, |
|
|
const double |
featureValue |
|
) |
| |
|
inline |
Create Split node and add it to certain tree
- Parameters
-
[in] | treeId | Tree to which new node is added |
[in] | parentId | Parent node to which new node is added (use noParent for root node) |
[in] | position | Position in parent (e.g. 0 for left and 1 for right child in a binary tree) |
[in] | featureIndex | Feature index for spliting |
[in] | featureValue | Feature value for spliting |
- Returns
- Node identifier
TreeId createTree |
( |
const size_t |
nNodes | ) |
|
|
inline |
Create certain tree in the decision forest model
- Parameters
-
[in] | nNodes | Number of nodes in created tree |
- Returns
- Tree identifier
Get built model
- Returns
- Model pointer
services::Status getStatus |
( |
| ) |
const |
|
inline |
Get status of model building
- Returns
- Status
const NodeId noParent = static_cast<size_t>(-1) |
|
static |
Reserved value for root nodes
The documentation for this class was generated from the following file: