xgboost
|
#include <quantile.h>
Public Types | |
using | WQSketch = WQuantileSketch< float, float > |
Public Member Functions | |
HostSketchContainer (std::vector< bst_row_t > columns_size, int32_t max_bins, bool use_group) | |
void | GatherSketchInfo (std::vector< WQSketch::SummaryContainer > const &reduced, std::vector< bst_row_t > *p_worker_segments, std::vector< bst_row_t > *p_sketches_scan, std::vector< WQSketch::Entry > *p_global_sketches) |
void | AllReduce (std::vector< WQSketch::SummaryContainer > *p_reduced, std::vector< int32_t > *p_num_cuts) |
void | PushRowPage (SparsePage const &page, MetaInfo const &info) |
void | MakeCuts (HistogramCuts *cuts) |
Static Public Member Functions | |
static bool | UseGroup (MetaInfo const &info) |
static std::vector< bst_row_t > | CalcColumnSize (SparsePage const &page, bst_feature_t const n_columns, size_t const nthreads) |
static std::vector< bst_feature_t > | LoadBalance (SparsePage const &page, bst_feature_t n_columns, size_t const nthreads) |
static uint32_t | SearchGroupIndFromRow (std::vector< bst_uint > const &group_ptr, size_t const base_rowid) |
A sketch matrix storing sketches for each feature.
using xgboost::common::HostSketchContainer::WQSketch = WQuantileSketch<float, float> |
xgboost::common::HostSketchContainer::HostSketchContainer | ( | std::vector< bst_row_t > | columns_size, |
int32_t | max_bins, | ||
bool | use_group | ||
) |
void xgboost::common::HostSketchContainer::AllReduce | ( | std::vector< WQSketch::SummaryContainer > * | p_reduced, |
std::vector< int32_t > * | p_num_cuts | ||
) |
|
static |
void xgboost::common::HostSketchContainer::GatherSketchInfo | ( | std::vector< WQSketch::SummaryContainer > const & | reduced, |
std::vector< bst_row_t > * | p_worker_segments, | ||
std::vector< bst_row_t > * | p_sketches_scan, | ||
std::vector< WQSketch::Entry > * | p_global_sketches | ||
) |
|
static |
void xgboost::common::HostSketchContainer::MakeCuts | ( | HistogramCuts * | cuts | ) |
void xgboost::common::HostSketchContainer::PushRowPage | ( | SparsePage const & | page, |
MetaInfo const & | info | ||
) |
|
inlinestatic |
|
inlinestatic |