Public Member Functions
void	Reset (Context const ctx, bst_bin_t total_bins, BatchParam const &p, bool is_distributed, bool is_col_split, HistMakerTrainParam const param)
	Reset the builder, should be called before growing a new tree.

template<bool any_missing>
void	BuildLocalHistograms (common::BlockedSpace2d const &space, GHistIndexMatrix const &gidx, std::vector< bst_node_t > const &nodes_to_build, common::RowSetCollection const &row_set_collection, common::Span< GradientPair const > gpair_h, bool force_read_by_column)

void	AddHistRows (RegTree const p_tree, std::vector< bst_node_t > p_nodes_to_build, std::vector< bst_node_t > *p_nodes_to_sub, bool rearrange)
	Allocate histogram, rearrange the nodes if `rearrange` is true and the tree has reached the cache size limit.

void	BuildHist (std::size_t page_idx, common::BlockedSpace2d const &space, GHistIndexMatrix const &gidx, common::RowSetCollection const &row_set_collection, std::vector< bst_node_t > const &nodes_to_build, linalg::VectorView< GradientPair const > gpair, bool force_read_by_column=false)
	Main entry point of this class, build histogram for tree nodes.

void	SyncHistogram (RegTree const *p_tree, std::vector< bst_node_t > const &nodes_to_build, std::vector< bst_node_t > const &nodes_to_trick)

BoundedHistCollection const &	Histogram () const

BoundedHistCollection &	Histogram ()

auto &	Buffer ()

Member Function Documentation

◆ Reset()

void xgboost::tree::HistogramBuilder::Reset	(	Context const *	ctx,
		bst_bin_t	total_bins,
		BatchParam const &	p,
		bool	is_distributed,
		bool	is_col_split,
		HistMakerTrainParam const *	param
	)

inline

Reset the builder, should be called before growing a new tree.

Parameters

total_bins	Total number of bins across all features
is_distributed	Mostly used for testing to allow injecting parameters instead of using global rabit variable.

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