Model Builder class for gradient boosted trees regression model.
More...
|
typedef size_t | NodeId |
| Node identifier type is size_t
|
|
typedef size_t | TreeId |
| Tree identifier type is size_t
|
|
Constructs the gradient boosted trees regression model builder
- Parameters
-
[in] | nIterations | Number of trees in model and iterations performed on training stage of gbt algorithm |
[in] | nFeatures | Number of features in training dataset |
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] | response | Response value for leaf node to be predicted |
- Returns
- Node identifier
NodeId addSplitNode |
( |
TreeId |
treeId, |
|
|
NodeId |
parentId, |
|
|
size_t |
position, |
|
|
size_t |
featureIndex, |
|
|
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 |
( |
size_t |
nNodes | ) |
|
|
inline |
Create certain tree in the gradient boosted trees regression model
- Parameters
-
[in] | nNodes | Number of nodes in created tree |
- Returns
- Tree identifier
Get built model
- Returns
- Model pointer
services::Status getStatus |
( |
| ) |
|
|
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: