Medial Code Documentation
Loading...
Searching...
No Matches
Public Member Functions | Protected Attributes
xgboost::tree::GloablApproxBuilder Class Reference

Public Member Functions

void InitData (DMatrix *p_fmat, RegTree const *p_tree, common::Span< float > hess)
 
CPUExpandEntry InitRoot (DMatrix *p_fmat, std::vector< GradientPair > const &gpair, common::Span< float > hess, RegTree *p_tree)
 
void UpdatePredictionCache (DMatrix const *data, linalg::MatrixView< float > out_preds) const
 
void BuildHistogram (DMatrix *p_fmat, RegTree *p_tree, std::vector< CPUExpandEntry > const &valid_candidates, std::vector< GradientPair > const &gpair, common::Span< float > hess)
 
void LeafPartition (RegTree const &tree, common::Span< float const > hess, std::vector< bst_node_t > *p_out_position)
 
 GloablApproxBuilder (TrainParam const *param, HistMakerTrainParam const *hist_param, MetaInfo const &info, Context const *ctx, std::shared_ptr< common::ColumnSampler > column_sampler, ObjInfo const *task, common::Monitor *monitor)
 
void UpdateTree (DMatrix *p_fmat, std::vector< GradientPair > const &gpair, common::Span< float > hess, RegTree *p_tree, HostDeviceVector< bst_node_t > *p_out_position)
 

Protected Attributes

TrainParam const * param_
 
HistMakerTrainParam const * hist_param_ {nullptr}
 
std::shared_ptr< common::ColumnSamplercol_sampler_
 
HistEvaluator evaluator_
 
MultiHistogramBuilder histogram_builder_
 
Context const * ctx_
 
ObjInfo const *const task_
 
std::vector< CommonRowPartitionerpartitioner_
 
RegTreep_last_tree_ {nullptr}
 
common::Monitormonitor_
 
size_t n_batches_ {0}
 
common::HistogramCuts feature_values_
 

Member Function Documentation

◆ UpdateTree()

void xgboost::tree::GloablApproxBuilder::UpdateTree ( DMatrix p_fmat,
std::vector< GradientPair > const &  gpair,
common::Span< float >  hess,
RegTree p_tree,
HostDeviceVector< bst_node_t > *  p_out_position 
)
inline

Note for update position Root: Not applied: No need to update position as initialization has got all the rows ordered. Applied: Update position is run on applied nodes so the rows are partitioned. Non-root: Not applied: That node is root of the subtree, same rule as root. Applied: Ditto


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