Medial Code Documentation
|
Public Types | |
using | WQSketch = WQuantileSketch< float, float > |
Public Member Functions | |
HostSketchContainer (Context const *ctx, bst_bin_t max_bins, common::Span< FeatureType const > ft, std::vector< size_t > columns_size, bool use_group) | |
template<typename Batch > | |
void | PushAdapterBatch (Batch const &batch, size_t base_rowid, MetaInfo const &info, float missing) |
![]() | |
SketchContainerImpl (Context const *ctx, std::vector< bst_row_t > columns_size, int32_t max_bins, common::Span< FeatureType const > feature_types, bool use_group) | |
void | GatherSketchInfo (MetaInfo const &info, std::vector< typename WQSketch::SummaryContainer > const &reduced, std::vector< bst_row_t > *p_worker_segments, std::vector< bst_row_t > *p_sketches_scan, std::vector< typename WQSketch::Entry > *p_global_sketches) |
void | AllReduce (MetaInfo const &info, std::vector< typename WQSketch::SummaryContainer > *p_reduced, std::vector< int32_t > *p_num_cuts) |
void | PushRowPageImpl (Batch const &batch, size_t base_rowid, OptionalWeights weights, size_t nnz, size_t n_features, bool is_dense, IsValid is_valid) |
void | PushRowPage (SparsePage const &page, MetaInfo const &info, Span< float const > hessian={}) |
void | MakeCuts (MetaInfo const &info, HistogramCuts *cuts) |
Additional Inherited Members | |
![]() | |
static bool | UseGroup (MetaInfo const &info) |
static uint32_t | SearchGroupIndFromRow (std::vector< bst_uint > const &group_ptr, size_t const base_rowid) |
![]() | |
std::vector< WQuantileSketch< float, float > > | sketches_ |
std::vector< std::set< float > > | categories_ |
std::vector< FeatureType > const | feature_types_ |
std::vector< bst_row_t > | columns_size_ |
int32_t | max_bins_ |
bool | use_group_ind_ |
int32_t | n_threads_ |
bool | has_categorical_ |
Monitor | monitor_ |