Medial Code Documentation
Loading...
Searching...
No Matches
Public Member Functions
xgboost::common::SortedSketchContainer Class Reference
Inheritance diagram for xgboost::common::SortedSketchContainer:
xgboost::common::SketchContainerImpl< WXQuantileSketch< float, float > >

Public Member Functions

 SortedSketchContainer (Context const *ctx, int32_t max_bins, common::Span< FeatureType const > ft, std::vector< size_t > columns_size, bool use_group)
 
void PushColPage (SparsePage const &page, MetaInfo const &info, Span< float const > hessian)
 Push a sorted CSC page.
 
- Public Member Functions inherited from xgboost::common::SketchContainerImpl< WXQuantileSketch< float, float > >
 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 Public Member Functions inherited from xgboost::common::SketchContainerImpl< WXQuantileSketch< float, float > >
static bool UseGroup (MetaInfo const &info)
 
static uint32_t SearchGroupIndFromRow (std::vector< bst_uint > const &group_ptr, size_t const base_rowid)
 
- Protected Attributes inherited from xgboost::common::SketchContainerImpl< WXQuantileSketch< float, float > >
std::vector< WXQuantileSketch< float, float > > 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_
 
int32_t n_threads_
 
bool has_categorical_
 
Monitor monitor_
 

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