xgboost
|
Cut configuration for sparse dataset. More...
#include <hist_util.h>
Public Member Functions | |
SparseCuts (HistogramCuts *container) | |
void | Concat (std::vector< std::unique_ptr< SparseCuts >> const &cuts, uint32_t n_cols) |
void | SingleThreadBuild (SparsePage const &page, MetaInfo const &info, uint32_t max_num_bins, bool const use_group_ind, uint32_t beg, uint32_t end, uint32_t thread_id) |
void | Build (DMatrix *dmat, uint32_t const max_num_bins) override |
![]() | |
CutsBuilder (HistogramCuts *p_cuts) | |
virtual | ~CutsBuilder ()=default |
void | AddCutPoint (WQSketch::SummaryContainer const &summary, int max_bin) |
Additional Inherited Members | |
![]() | |
using | WQSketch = common::WQuantileSketch< bst_float, bst_float > |
![]() | |
static bool | UseGroup (DMatrix *dmat) |
static uint32_t | SearchGroupIndFromRow (std::vector< bst_uint > const &group_ptr, size_t const base_rowid) |
![]() | |
HistogramCuts * | p_cuts_ |
Cut configuration for sparse dataset.
|
inlineexplicit |
|
overridevirtual |
Implements xgboost::common::CutsBuilder.
void xgboost::common::SparseCuts::Concat | ( | std::vector< std::unique_ptr< SparseCuts >> const & | cuts, |
uint32_t | n_cols | ||
) |
void xgboost::common::SparseCuts::SingleThreadBuild | ( | SparsePage const & | page, |
MetaInfo const & | info, | ||
uint32_t | max_num_bins, | ||
bool const | use_group_ind, | ||
uint32_t | beg, | ||
uint32_t | end, | ||
uint32_t | thread_id | ||
) |