StochTree 0.0.1
Loading...
Searching...
No Matches
Public Member Functions | Public Attributes | List of all members
StochTree::NodeCutpointTracker Class Reference

Computing and tracking cutpoints available for a given feature at a given node. More...

#include <cutpoint_candidates.h>

Public Member Functions

void CalculateStrides (Eigen::MatrixXd &covariates, Eigen::VectorXd &residuals, SortedNodeSampleTracker *feature_node_sort_tracker, int32_t node_id, data_size_t node_begin, data_size_t node_end, int32_t feature_index)
 Calculate strides.
 
void CalculateStridesNumeric (Eigen::MatrixXd &covariates, Eigen::VectorXd &residuals, SortedNodeSampleTracker *feature_node_sort_tracker, data_size_t node_begin, data_size_t node_end, int32_t feature_index)
 Split numeric / ordered categorical feature and update sort indices.
 
void CalculateStridesCategorical (Eigen::MatrixXd &covariates, Eigen::VectorXd &residuals, SortedNodeSampleTracker *feature_node_sort_tracker, data_size_t node_begin, data_size_t node_end, int32_t feature_index)
 Split unordered categorical feature and update sort indices.
 
int32_t NumCutpoints ()
 Number of potential cutpoints enumerated.
 
bool NodeCutpointEvaluated (int32_t node_id)
 Whether a cutpoint grid has been enumerated for a given node.
 
int32_t CurrentNodeEvaluated ()
 Node id of the node that has been most recently evaluated.
 

Public Attributes

std::vector< data_size_t > node_stride_begin_
 Vectors of node stride starting points and stride lengths.
 

Detailed Description

Computing and tracking cutpoints available for a given feature at a given node.


The documentation for this class was generated from the following file: