Data structure for tracking observations through a tree partition with each feature pre-sorted. More...

Public Member Functions
void	PartitionNode (Eigen::MatrixXd &covariates, int node_id, int feature_split, TreeSplit &split, int num_threads=-1)
	Partition a node based on a new split rule.

void	PartitionNode (Eigen::MatrixXd &covariates, int node_id, int feature_split, double split_value, int num_threads=-1)
	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, int num_threads=-1)
	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.

Detailed Description

Data structure for tracking observations through a tree partition with each feature pre-sorted.

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

Public Member Functions