StochTree::FeatureUnsortedPartition Class Reference

Mapping nodes to the indices they contain. More...

#include <partition_tracker.h>

Public Member Functions
void	ReconstituteFromTree (Tree &tree, ForestDataset &dataset)
	Reconstitute a tree partition tracker from root based on a tree.
void	PartitionNode (Eigen::MatrixXd &covariates, int node_id, int left_node_id, int right_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 left_node_id, int right_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 left_node_id, int right_node_id, int feature_split, std::vector< std::uint32_t > const &category_list)
	Partition a node based on a new split rule.
void	PruneNodeToLeaf (int node_id)
	Convert a (currently split) node to a leaf.
bool	IsLeaf (int node_id)
	Whether node_id is a leaf.
bool	IsValidNode (int node_id)
	Whether node_id is a valid node.
bool	LeftNodeIsLeaf (int node_id)
	Whether node_id's left child is a leaf.
bool	RightNodeIsLeaf (int node_id)
	Whether node_id's right child is a leaf.
data_size_t	NodeBegin (int node_id)
	First index of data points contained in node_id.
data_size_t	NodeEnd (int node_id)
	One past the last index of data points contained in node_id.
data_size_t	NodeSize (int node_id)
	Number of data points contained in node_id.
int	Parent (int node_id)
	Parent node_id.
int	LeftNode (int node_id)
	Left child of node_id.
int	RightNode (int node_id)
	Right child of node_id.
std::vector< data_size_t >	NodeIndices (int node_id)
	Data indices for a given node.
void	UpdateObservationMapping (int node_id, int tree_id, SampleNodeMapper *sample_node_mapper)
	Update SampleNodeMapper for all the observations in node_id.

Public Attributes
std::vector< data_size_t >	indices_
	Data indices.

Detailed Description

Mapping nodes to the indices they contain.

---

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

• include/stochtree/partition_tracker.h