|
StochTree 0.2.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.