StochTree 0.0.1
|
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. | |
Computing and tracking cutpoints available for a given feature at a given node.