Medial Code Documentation
|
Histogram builder that can handle multiple targets. More...
#include <histogram.h>
Public Member Functions | |
template<typename Partitioner , typename ExpandEntry > | |
void | BuildRootHist (DMatrix *p_fmat, RegTree const *p_tree, std::vector< Partitioner > const &partitioners, linalg::MatrixView< GradientPair const > gpair, ExpandEntry const &best, BatchParam const ¶m, bool force_read_by_column=false) |
Build the histogram for root node. | |
template<typename Partitioner , typename ExpandEntry > | |
void | BuildHistLeftRight (DMatrix *p_fmat, RegTree const *p_tree, std::vector< Partitioner > const &partitioners, std::vector< ExpandEntry > const &valid_candidates, linalg::MatrixView< GradientPair const > gpair, BatchParam const ¶m, bool force_read_by_column=false) |
Build histogram for left and right child of valid candidates. | |
auto const & | Histogram (bst_target_t t) const |
auto & | Histogram (bst_target_t t) |
void | Reset (Context const *ctx, bst_bin_t total_bins, bst_target_t n_targets, BatchParam const &p, bool is_distributed, bool is_col_split, HistMakerTrainParam const *param) |
Histogram builder that can handle multiple targets.