Public Member Functions
	Builder (const TrainParam &param, const ColMakerTrainParam &colmaker_train_param, FeatureInteractionConstraintHost _interaction_constraints, Context const *ctx, const std::vector< float > &column_densities)

virtual void	Update (const std::vector< GradientPair > &gpair, DMatrix p_fmat, RegTree p_tree)

Protected Member Functions
void	InitData (const std::vector< GradientPair > &gpair, const DMatrix &fmat)

void	InitNewNode (const std::vector< int > &qexpand, const std::vector< GradientPair > &gpair, const DMatrix &fmat, const RegTree &tree)
	initialize the base_weight, root_gain, and NodeEntry for all the new nodes in qexpand

void	UpdateQueueExpand (const RegTree &tree, const std::vector< int > &qexpand, std::vector< int > *p_newnodes)
	update queue expand add in new leaves

void	UpdateEnumeration (int nid, GradientPair gstats, bst_float fvalue, int d_step, bst_uint fid, GradStats &c, std::vector< ThreadEntry > &temp, TreeEvaluator::SplitEvaluator< TrainParam > const &evaluator) const

void	EnumerateSplit (const Entry begin, const Entry end, int d_step, bst_uint fid, const std::vector< GradientPair > &gpair, std::vector< ThreadEntry > &temp, TreeEvaluator::SplitEvaluator< TrainParam > const &evaluator) const

virtual void	UpdateSolution (const SortedCSCPage &batch, const std::vector< bst_feature_t > &feat_set, const std::vector< GradientPair > &gpair, DMatrix *)

void	FindSplit (int depth, const std::vector< int > &qexpand, const std::vector< GradientPair > &gpair, DMatrix p_fmat, RegTree p_tree)

void	ResetPosition (const std::vector< int > &qexpand, DMatrix *p_fmat, const RegTree &tree)

virtual void	SyncBestSolution (const std::vector< int > &qexpand)

virtual void	SetNonDefaultPosition (const std::vector< int > &qexpand, DMatrix *p_fmat, const RegTree &tree)

int	DecodePosition (bst_uint ridx) const

void	SetEncodePosition (bst_uint ridx, int nid)

Protected Attributes
const TrainParam &	param_

const ColMakerTrainParam &	colmaker_train_param_

Context const *	ctx_

common::ColumnSampler	column_sampler_

std::vector< int >	position_

std::vector< std::vector< ThreadEntry > >	stemp_

std::vector< NodeEntry >	snode_
	TreeNode Data: statistics for each constructed node.

std::vector< int >	qexpand_
	queue of nodes to be expanded

TreeEvaluator	tree_evaluator_

FeatureInteractionConstraintHost	interaction_constraints_

const std::vector< float > &	column_densities_

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

External/xgboost/src/tree/updater_colmaker.cc

Public Member Functions

Protected Member Functions

Protected Attributes