Medial Code Documentation
Loading...
Searching...
No Matches
Public Member Functions | Static Public Member Functions | Data Fields | Protected Member Functions
xgboost::common::HistogramCuts Class Reference

Public Member Functions

 HistogramCuts (HistogramCuts const &that)
 
 HistogramCuts (HistogramCuts &&that) noexcept(true)
 
HistogramCutsoperator= (HistogramCuts const &that)
 
HistogramCutsoperator= (HistogramCuts &&that) noexcept(true)
 
bst_bin_t FeatureBins (bst_feature_t feature) const
 
std::vector< uint32_t > const & Ptrs () const
 
std::vector< float > const & Values () const
 
std::vector< float > const & MinValues () const
 
bool HasCategorical () const
 
float MaxCategory () const
 
void SetCategorical (bool has_cat, float max_cat)
 Set meta info about categorical features.
 
bst_bin_t TotalBins () const
 
bst_bin_t SearchBin (float value, bst_feature_t column_id, std::vector< uint32_t > const &ptrs, std::vector< float > const &values) const
 
bst_bin_t SearchBin (float value, bst_feature_t column_id) const
 
bst_bin_t SearchBin (Entry const &e) const
 Search the bin index for numerical feature.
 
bst_bin_t SearchCatBin (float value, bst_feature_t fidx, std::vector< uint32_t > const &ptrs, std::vector< float > const &vals) const
 Search the bin index for categorical feature.
 
bst_bin_t SearchCatBin (float value, bst_feature_t fidx) const
 
bst_bin_t SearchCatBin (Entry const &e) const
 

Static Public Member Functions

static float NumericBinValue (std::vector< std::uint32_t > const &ptrs, std::vector< float > const &vals, std::vector< float > const &mins, bst_feature_t fidx, bst_bin_t bin_idx)
 Return numerical bin value given bin index.
 

Data Fields

HostDeviceVector< float > cut_values_
 
HostDeviceVector< uint32_t > cut_ptrs_
 
HostDeviceVector< float > min_vals_
 

Protected Member Functions

void Swap (HistogramCuts &&that) noexcept(true)
 
void Copy (HistogramCuts const &that)
 

Member Function Documentation

◆ SetCategorical()

void xgboost::common::HistogramCuts::SetCategorical ( bool  has_cat,
float  max_cat 
)
inline

Set meta info about categorical features.

Parameters
has_catDo we have categorical feature in the data?
max_catThe maximum categorical value in all features.

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