|
StochTree 0.0.1
|
Data structure for tracking observations through a tree partition with each feature pre-sorted. More...
#include <partition_tracker.h>
Public Member Functions | |
| void | PartitionNode (Eigen::MatrixXd &covariates, int node_id, int feature_split, TreeSplit &split) |
| Partition a node based on a new split rule. | |
| void | PartitionNode (Eigen::MatrixXd &covariates, int node_id, int feature_split, double split_value) |
| Partition a node based on a new split rule. | |
| void | PartitionNode (Eigen::MatrixXd &covariates, int node_id, int feature_split, std::vector< std::uint32_t > const &category_list) |
| Partition a node based on a new split rule. | |
| data_size_t | NodeBegin (int node_id, int feature_index) |
| First index of data points contained in node_id. | |
| data_size_t | NodeEnd (int node_id, int feature_index) |
| One past the last index of data points contained in node_id. | |
| data_size_t | NodeSize (int node_id, int feature_index) |
| One past the last index of data points contained in node_id. | |
| std::vector< data_size_t > | NodeIndices (int node_id, int feature_index) |
| Data indices for a given node. | |
| data_size_t | SortIndex (data_size_t j, int feature_index) |
| Feature sort index j for feature_index. | |
| void | UpdateObservationMapping (int node_id, int tree_id, SampleNodeMapper *sample_node_mapper, int feature_index=0) |
| Update SampleNodeMapper for all the observations in node_id. | |
Data structure for tracking observations through a tree partition with each feature pre-sorted.
1.9.8