#include <quantile.h>
|
| HostSketchContainer (int32_t max_bins, MetaInfo const &info, std::vector< size_t > columns_size, bool use_group, Span< float const > hessian, int32_t n_threads) |
|
| SketchContainerImpl (std::vector< bst_row_t > columns_size, int32_t max_bins, common::Span< FeatureType const > feature_types, bool use_group, int32_t n_threads) |
|
void | GatherSketchInfo (std::vector< typename WQuantileSketch< float, float > ::SummaryContainer > const &reduced, std::vector< bst_row_t > *p_worker_segments, std::vector< bst_row_t > *p_sketches_scan, std::vector< typename WQuantileSketch< float, float > ::Entry > *p_global_sketches) |
|
void | AllReduce (std::vector< typename WQuantileSketch< float, float > ::SummaryContainer > *p_reduced, std::vector< int32_t > *p_num_cuts) |
|
void | PushRowPage (SparsePage const &page, MetaInfo const &info, Span< float const > hessian={}) |
|
void | MakeCuts (HistogramCuts *cuts) |
|
◆ WQSketch
◆ HostSketchContainer()
xgboost::common::HostSketchContainer::HostSketchContainer |
( |
int32_t |
max_bins, |
|
|
MetaInfo const & |
info, |
|
|
std::vector< size_t > |
columns_size, |
|
|
bool |
use_group, |
|
|
Span< float const > |
hessian, |
|
|
int32_t |
n_threads |
|
) |
| |
The documentation for this class was generated from the following file:
- /home/runner/work/xgboost/xgboost/src/common/quantile.h