C++ API Reference for Intel® Data Analytics Acceleration Library 2020 Update 1

Public Types | Static Public Attributes | List of all members
ModelBuilder Class Reference

Model Builder class for gradient boosted trees classification model. More...

Public Types

typedef size_t NodeId
 Node identifier type is size_t
 
typedef size_t TreeId
 Tree identifier type is size_t
 

Static Public Attributes

static const NodeId noParent = static_cast<size_t>(-1)
 

Class Declaration

References

Constructor & Destructor Documentation

ModelBuilder ( size_t  nFeatures,
size_t  nIterations,
size_t  nClasses = 2 
)
inline

Constructs the gradient boosted trees classification model builder

Parameters
[in]nFeaturesNumber of features in training data
[in]nIterationsNumber of trees in model for each class
[in]nClassesNumber of classes in model

Member Function Documentation

NodeId addLeafNode ( TreeId  treeId,
NodeId  parentId,
size_t  position,
double  response 
)
inline

Create Leaf node and add it to certain tree

Parameters
[in]treeIdTree to which new node is added
[in]parentIdParent node to which new node is added (use noParent for root node)
[in]positionPosition in parent (e.g. 0 for left and 1 for right child in a binary tree)
[in]responseResponse 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]treeIdTree to which new node is added
[in]parentIdParent node to which new node is added (use noParent for root node)
[in]positionPosition in parent (e.g. 0 for left and 1 for right child in a binary tree)
[in]featureIndexFeature index for spliting
[in]featureValueFeature value for spliting
Returns
Node identifier
TreeId createTree ( size_t  nNodes,
size_t  classLabel 
)
inline

Create certain tree in the gradient boosted trees classification model for certain class

Parameters
[in]nNodesNumber of nodes in created tree
[in]classLabelLabel of class for which tree is created. classLabel bellows interval from 0 to (nClasses - 1)
Returns
Tree identifier
ModelPtr getModel ( )
inline

Get built model

Returns
Model pointer
services::Status getStatus ( )
inline

Get status of model building

Returns
Status

Member Data Documentation

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:

For more complete information about compiler optimizations, see our Optimization Notice.