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

Updater for building multi-target trees. More...

Public Member Functions

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

Detailed Description

Updater for building multi-target trees.

The implementation simply iterates over each target.


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