Medial Code Documentation
|
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) |
Tree updater for single-target trees.
|
inline |
Specialized code for dense data: For dense data (with no missing value), the sum of gradient histogram is equal to snode[nid]