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

Container class for FeatureCutpointGrid objects stored for every feature in a dataset. 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, std::vector< FeatureType > &feature_types)
 Calculate strides.
 
int32_t CutpointGridSize ()
 Max size of cutpoint grid.
 
int32_t NumCutpoints (int feature_index)
 Number of potential cutpoints enumerated.
 
int32_t BinStartIndex (int i, int feature_index)
 Beginning index of bin i.
 
int32_t BinLength (int i, int feature_index)
 Size of bin i.
 
int32_t BinEndIndex (int i, int feature_index)
 Beginning index of bin i.
 
double CutpointValue (int i, int feature_index)
 Value of the upper-bound (cutpoint) implied by bin i.
 
std::vector< std::uint32_t > CutpointVector (int i, int feature_index)
 Vector of cutpoint values up to and including bin i Helper function for converting categorical split "value" (as outlined in Fisher 1958) to a set of categories.
 

Detailed Description

Container class for FeatureCutpointGrid objects stored for every feature in a dataset.


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