|
struct | AFTLoss |
| The AFT loss function. More...
|
|
struct | AFTParam |
| Parameter structure for AFT loss and metric. More...
|
|
class | Base64InStream |
| the stream that reads from base64, note we take from file pointers More...
|
|
class | Base64OutStream |
| the stream that write to base64, note we take from file pointers More...
|
|
class | BlockedSpace2d |
|
class | Column |
| a column storage, to be used with ApplySplit. Note that each bin id is stored as index[i] + index_base. Different types of column index for each column allow to reduce the memory usage. More...
|
|
class | ColumnMatrix |
| a collection of columns, with support for construction from GHistIndexMatrix. More...
|
|
class | ColumnSampler |
| Handles selection of columns due to colsample_bytree, colsample_bylevel and colsample_bynode parameters. Should be initialised before tree construction and to reset when tree construction is completed. More...
|
|
class | CompressedBufferWriter |
| Writes bit compressed symbols to a memory buffer. Use CompressedIterator to read symbols back from buffer. Currently limited to a maximum symbol size of 28 bits. More...
|
|
class | CompressedIterator |
| Read symbols from a bit compressed memory buffer. Usable on device and host. More...
|
|
class | ConfigParser |
| Implementation of config reader. More...
|
|
class | DenseColumn |
|
struct | ExtremeDistribution |
|
class | FixedSizeStream |
| A simple class used to consume ‘dmlc::Stream’ all at once. More...
|
|
class | GHistBuilder |
| builder for histograms of gradient statistics More...
|
|
class | HistCollection |
| histogram of gradient statistics for multiple nodes More...
|
|
class | HistogramCuts |
|
class | HostSketchContainer |
|
struct | Index |
| Optionally compressed gradient index. The compression works only with dense data. More...
|
|
class | IndexTransformIter |
| Transform iterator that takes an index and calls transform operator. More...
|
|
struct | IsCatOp |
|
struct | LogisticDistribution |
|
class | MemStackAllocator |
| A C-style array with in-stack allocation. As long as the array is smaller than MaxStackSize, it will be allocated inside the stack. Otherwise, it will be heap-allocated. More...
|
|
struct | Monitor |
| Timing utility used to measure total method execution time over the lifetime of the containing object. More...
|
|
struct | NormalDistribution |
|
struct | OptionalWeights |
|
class | ParallelGHistBuilder |
| Stores temporary histograms to compute them in parallel Supports processing multiple tree-nodes for nested parallelism Able to reduce histograms across threads in efficient way. More...
|
|
class | ParallelGroupBuilder |
| multi-thread version of group builder More...
|
|
class | PartitionBuilder |
|
class | PeekableInStream |
| Input stream that support additional PeekRead operation, besides read. More...
|
|
class | QuantileSketchTemplate |
| template for all quantile sketch algorithm that uses merge/prune scheme More...
|
|
class | Range |
|
class | Range1d |
|
class | RowSetCollection |
| collection of rowset More...
|
|
struct | Sched |
|
class | SketchContainerImpl |
|
struct | SortedQuantile |
| Quantile structure accepts sorted data, extracted from histmaker. More...
|
|
class | SortedSketchContainer |
|
class | Span |
| span class implementation, based on ISO++20 span<T>. The interface should be the same. More...
|
|
class | SparseColumn |
|
class | StreamBufferReader |
| buffer reader of the stream that allows you to get More...
|
|
struct | Timer |
|
class | Transform |
| Do Transformation on HostDeviceVectors. More...
|
|
struct | WQSummary |
| experimental wsummary More...
|
|
class | WQuantileSketch |
| Quantile sketch use WQSummary. More...
|
|
struct | WXQSummary |
| try to do efficient pruning More...
|
|
class | WXQuantileSketch |
| Quantile sketch use WXQSummary. More...
|
|
|
template<class T , std::size_t X, class U , std::size_t Y> |
XGBOOST_DEVICE bool | operator== (Span< T, X > l, Span< U, Y > r) |
|
template<class T , std::size_t X, class U , std::size_t Y> |
constexpr XGBOOST_DEVICE bool | operator!= (Span< T, X > l, Span< U, Y > r) |
|
template<class T , std::size_t X, class U , std::size_t Y> |
constexpr XGBOOST_DEVICE bool | operator< (Span< T, X > l, Span< U, Y > r) |
|
template<class T , std::size_t X, class U , std::size_t Y> |
constexpr XGBOOST_DEVICE bool | operator<= (Span< T, X > l, Span< U, Y > r) |
|
template<class T , std::size_t X, class U , std::size_t Y> |
constexpr XGBOOST_DEVICE bool | operator> (Span< T, X > l, Span< U, Y > r) |
|
template<class T , std::size_t X, class U , std::size_t Y> |
constexpr XGBOOST_DEVICE bool | operator>= (Span< T, X > l, Span< U, Y > r) |
|
template<class T , std::size_t E> |
XGBOOST_DEVICE auto | as_bytes (Span< T, E > s) __span_noexcept -> Span< const byte, detail::ExtentAsBytesValue< T, E >::value > |
|
template<class T , std::size_t E> |
XGBOOST_DEVICE auto | as_writable_bytes (Span< T, E > s) __span_noexcept -> Span< byte, detail::ExtentAsBytesValue< T, E >::value > |
|
template<typename T > |
XGBOOST_DEVICE bst_cat_t | AsCat (T const &v) |
|
XGBOOST_DEVICE bool | IsCat (Span< FeatureType const > ft, bst_feature_t fidx) |
|
constexpr bst_cat_t | OutOfRangeCat () |
|
XGBOOST_DEVICE bool | InvalidCat (float cat) |
|
template<bool validate = true> |
XGBOOST_DEVICE bool | Decision (common::Span< uint32_t const > cats, float cat, bool dft_left) |
|
void | InvalidCategory () |
|
void | CheckMaxCat (float max_cat, size_t n_categories) |
|
XGBOOST_DEVICE bool | UseOneHot (uint32_t n_cats, uint32_t max_cat_to_onehot) |
| Whether should we use onehot encoding for categorical data. More...
|
|
std::vector< std::string > | Split (const std::string &s, char delim) |
| Split a string by delimiter. More...
|
|
template<typename T > |
XGBOOST_DEVICE T | Max (T a, T b) |
|
template<typename T > |
std::string | ToString (const T &data) |
|
template<typename T1 , typename T2 > |
XGBOOST_DEVICE T1 | DivRoundUp (const T1 a, const T2 b) |
|
template<class T , std::size_t N> |
constexpr auto | UnpackArr (std::array< T, N > &&arr) |
|
template<typename Fn > |
auto | MakeIndexTransformIter (Fn &&fn) |
|
int | AllVisibleGPUs () |
|
void | AssertGPUSupport () |
|
void | AssertOneAPISupport () |
|
template<typename Idx , typename Container , typename V = typename Container::value_type, typename Comp = std::less<V>> |
std::vector< Idx > | ArgSort (Container const &array, Comp comp=std::less< V >{}) |
|
HistogramCuts | SketchOnDMatrix (DMatrix *m, int32_t max_bins, int32_t n_threads, bool use_sorted=false, Span< float > const hessian={}) |
| Run CPU sketching on DMatrix. More...
|
|
template<typename GradientIndex > |
int32_t XGBOOST_HOST_DEV_INLINE | BinarySearchBin (size_t begin, size_t end, GradientIndex const &data, uint32_t const fidx_begin, uint32_t const fidx_end) |
|
template<typename GradientSumT > |
void | InitilizeHistByZeroes (GHistRow< GradientSumT > hist, size_t begin, size_t end) |
| fill a histogram by zeros More...
|
|
template<typename GradientSumT > |
void | IncrementHist (GHistRow< GradientSumT > dst, const GHistRow< GradientSumT > add, size_t begin, size_t end) |
| Increment hist as dst += add in range [begin, end) More...
|
|
template<typename GradientSumT > |
void | CopyHist (GHistRow< GradientSumT > dst, const GHistRow< GradientSumT > src, size_t begin, size_t end) |
| Copy hist from src to dst in range [begin, end) More...
|
|
template<typename GradientSumT > |
void | SubtractionHist (GHistRow< GradientSumT > dst, const GHistRow< GradientSumT > src1, const GHistRow< GradientSumT > src2, size_t begin, size_t end) |
| Compute Subtraction: dst = src1 - src2 in range [begin, end) More...
|
|
std::string | LoadSequentialFile (std::string uri, bool stream=false) |
| Helper function for loading consecutive file to avoid dmlc Stream when possible. More...
|
|
std::string | FileExtension (std::string fname, bool lower=true) |
| Get file extension from file name. More...
|
|
std::string | ReadAll (dmlc::Stream *fi, PeekableInStream *fp) |
| Read the whole buffer from dmlc stream. More...
|
|
XGBOOST_DEVICE float | Sigmoid (float x) |
| calculate the sigmoid of the input. More...
|
|
template<typename T , typename U > |
constexpr XGBOOST_DEVICE bool | CloseTo (T a, U b) |
| Equality test for both integer and floating point. More...
|
|
template<typename Iterator > |
XGBOOST_DEVICE void | Softmax (Iterator start, Iterator end) |
| Do inplace softmax transformaton on start to end. More...
|
|
template<typename Iterator > |
XGBOOST_DEVICE Iterator | FindMaxIndex (Iterator begin, Iterator end) |
| Find the maximum iterator within the iterators. More...
|
|
float | LogSum (float x, float y) |
| perform numerically safe logsum More...
|
|
template<typename Iterator > |
float | LogSum (Iterator begin, Iterator end) |
| perform numerically safe logsum More...
|
|
template<typename T > |
XGBOOST_DEVICE std::enable_if< std::numeric_limits< T >::is_integer, bool >::type | CheckNAN (T) |
|
XGBOOST_DEVICE bool | CheckNAN (float x) |
|
XGBOOST_DEVICE bool | CheckNAN (double x) |
|
template<typename T > |
XGBOOST_DEVICE T | LogGamma (T v) |
|
GlobalRandomEngine & | GlobalRandom () |
| global singleton of a random engine. This random engine is thread-local and only visible to current thread. More...
|
|
template<typename T > |
std::vector< T > | WeightedSamplingWithoutReplacement (std::vector< T > const &array, std::vector< float > const &weights, size_t n) |
|
template<typename Func > |
void | ParallelFor2d (const BlockedSpace2d &space, int nthreads, Func func) |
|
template<typename Index , typename Func > |
void | ParallelFor (Index size, int32_t n_threads, Sched sched, Func fn) |
|
template<typename Index , typename Func > |
void | ParallelFor (Index size, int32_t n_threads, Func fn) |
|
int32_t | OmpGetThreadLimit () |
|
int32_t | GetCfsCPUCount () noexcept |
|
int32_t | OmpGetNumThreads (int32_t n_threads) |
|