Medial Code Documentation
|
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 |
Updater for building multi-target trees.
The implementation simply iterates over each target.