Feature interaction constraint implementation for CPU tree updaters. More...

Public Member Functions
void	Split (int32_t node_id, bst_feature_t feature_id, bst_node_t left_id, bst_node_t right_id)

bool	Query (bst_node_t nid, bst_feature_t fid) const

void	Reset ()

void	Configure (tree::TrainParam const &param, bst_feature_t const n_features)

Protected Member Functions
void	SplitImpl (int32_t node_id, bst_feature_t feature_id, bst_node_t left_id, bst_node_t right_id)

Protected Attributes
std::vector< std::unordered_set< bst_feature_t > >	interaction_constraints_

std::vector< std::unordered_set< bst_feature_t > >	node_constraints_

std::vector< std::unordered_set< bst_feature_t > >	splits_

std::string	interaction_constraint_str_

bst_feature_t	n_features_

bool	enabled_ {false}

Detailed Description

Feature interaction constraint implementation for CPU tree updaters.

The interface is similar to the one for GPU Hist.

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

Public Member Functions