Medial Code Documentation
Loading...
Searching...
No Matches
Public Member Functions | Static Public Member Functions | Protected Attributes
xgboost::common::SketchContainerImpl< WQSketch > Class Template Reference

#include <quantile.h>

Public Member Functions

 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)
 
template<typename Batch , typename IsValid >
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)
 

Static Public Member Functions

static bool UseGroup (MetaInfo const &info)
 
static uint32_t SearchGroupIndFromRow (std::vector< bst_uint > const &group_ptr, size_t const base_rowid)
 

Protected Attributes

std::vector< WQSketch > sketches_
 
std::vector< std::set< float > > categories_
 
std::vector< FeatureType > const feature_types_
 
std::vector< bst_row_tcolumns_size_
 
int32_t max_bins_
 
bool use_group_ind_ {false}
 
int32_t n_threads_
 
bool has_categorical_ {false}
 
Monitor monitor_
 

Detailed Description

template<typename WQSketch>
class xgboost::common::SketchContainerImpl< WQSketch >

A sketch matrix storing sketches for each feature.


The documentation for this class was generated from the following files: