Class for building model of the gradient boosted trees regression algorithm
More...
Constructs the gradient boosted trees regression model builder
- Parameters
-
context | Context to manage gradient boosted trees regression model builder |
nFeatures | Number of features in training data |
nIterations | Number of trees in model and iterations performed on training stage of gbt algorithm |
long addLeafNode |
( |
long |
treeId, |
|
|
long |
parentId, |
|
|
long |
position, |
|
|
double |
response |
|
) |
| |
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) |
response | Response value for leaf node 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 | ) |
|
Create certain tree in the gradient boosted trees regression model
- Parameters
-
nNodes | Number of nodes in created tree |
- Returns
- Positive number tree identifier
Get built model of gradient boosted trees regression
- Returns
- Model of gradient boosted trees regression
The documentation for this class was generated from the following file:
- gbt/regression/ModelBuilder.java