Medial Code Documentation
Loading...
Searching...
No Matches
Public Member Functions
xgboost::tree::HistUpdater Class Reference

Tree updater for single-target trees. More...

Public Member Functions

 HistUpdater (Context const *ctx, std::shared_ptr< common::ColumnSampler > column_sampler, TrainParam const *param, HistMakerTrainParam const *hist_param, DMatrix const *fmat, ObjInfo const *task, common::Monitor *monitor)
 
bool UpdatePredictionCache (DMatrix const *data, linalg::MatrixView< float > out_preds) const
 
void InitData (DMatrix *fmat, RegTree const *p_tree)
 
void EvaluateSplits (DMatrix *p_fmat, RegTree const *p_tree, std::vector< CPUExpandEntry > *best_splits)
 
void ApplyTreeSplit (CPUExpandEntry const &candidate, RegTree *p_tree)
 
CPUExpandEntry InitRoot (DMatrix *p_fmat, linalg::MatrixView< GradientPair const > gpair, RegTree *p_tree)
 
void BuildHistogram (DMatrix *p_fmat, RegTree *p_tree, std::vector< CPUExpandEntry > const &valid_candidates, linalg::MatrixView< GradientPair const > gpair)
 
void UpdatePosition (DMatrix *p_fmat, RegTree const *p_tree, std::vector< CPUExpandEntry > const &applied)
 
void LeafPartition (RegTree const &tree, linalg::MatrixView< GradientPair const > gpair, std::vector< bst_node_t > *p_out_position)
 

Detailed Description

Tree updater for single-target trees.

Member Function Documentation

◆ InitRoot()

CPUExpandEntry xgboost::tree::HistUpdater::InitRoot ( DMatrix p_fmat,
linalg::MatrixView< GradientPair const >  gpair,
RegTree p_tree 
)
inline

Specialized code for dense data: For dense data (with no missing value), the sum of gradient histogram is equal to snode[nid]


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