xgboost
|
Namespaces | |
aft | |
base64 | |
namespace of base64 decoding and encoding table | |
detail | |
Classes | |
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... | |
struct | GHistIndexBlock |
class | GHistIndexBlockMatrix |
struct | GHistIndexMatrix |
preprocessed global index matrix, in CSR format More... | |
class | HistCollection |
histogram of gradient statistics for multiple nodes More... | |
class | HistogramCuts |
class | HostSketchContainer |
struct | Index |
struct | LogisticDistribution |
struct | Monitor |
Timing utility used to measure total method execution time over the lifetime of the containing object. More... | |
struct | NormalDistribution |
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... | |
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... | |
Typedefs | |
using | CompressedByteT = unsigned char |
using | GHistIndexRow = Span< uint32_t const > |
A single row in global histogram index. Directly represent the global index in the histogram entry. More... | |
template<typename GradientSumT > | |
using | GHistRow = Span< xgboost::detail::GradientPairInternal< GradientSumT > > |
using | MemoryFixSizeBuffer = rabit::utils::MemoryFixSizeBuffer |
using | MemoryBufferStream = rabit::utils::MemoryBufferStream |
using | RandomEngine = std::mt19937 |
Define mt19937 as default type Random Engine. More... | |
using | GlobalRandomEngine = RandomEngine |
global random engine More... | |
Enumerations | |
enum | byte : unsigned char |
enum | ColumnType { kDenseColumn, kSparseColumn } |
column type More... | |
enum | BinTypeSize { kUint8BinsTypeSize = 1, kUint16BinsTypeSize = 2, kUint32BinsTypeSize = 4 } |
enum | ProbabilityDistributionType : int { ProbabilityDistributionType::kNormal = 0, ProbabilityDistributionType::kLogistic = 1, ProbabilityDistributionType::kExtreme = 2 } |
Enum encoding possible choices of probability distribution. More... | |
enum | CensoringType : uint8_t { CensoringType::kUncensored, CensoringType::kRightCensored, CensoringType::kLeftCensored, CensoringType::kIntervalCensored } |
Functions | |
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> | |
XGBOOST_DEVICE constexpr bool | operator!= (Span< T, X > l, Span< U, Y > r) |
template<class T , std::size_t X, class U , std::size_t Y> | |
XGBOOST_DEVICE constexpr bool | operator< (Span< T, X > l, Span< U, Y > r) |
template<class T , std::size_t X, class U , std::size_t Y> | |
XGBOOST_DEVICE constexpr bool | operator<= (Span< T, X > l, Span< U, Y > r) |
template<class T , std::size_t X, class U , std::size_t Y> | |
XGBOOST_DEVICE constexpr bool | operator> (Span< T, X > l, Span< U, Y > r) |
template<class T , std::size_t X, class U , std::size_t Y> | |
XGBOOST_DEVICE constexpr 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 > |
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) |
int | AllVisibleGPUs () |
void | AssertGPUSupport () |
void | AssertOneAPISupport () |
template<typename Idx , typename V , typename Comp = std::less<V>> | |
std::vector< Idx > | ArgSort (std::vector< V > const &array, Comp comp=std::less< V >{}) |
HistogramCuts | SketchOnDMatrix (DMatrix *m, int32_t max_bins) |
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 const &fname) |
XGBOOST_DEVICE float | Sigmoid (float x) |
calculate the sigmoid of the input. More... | |
template<typename T , typename U > | |
XGBOOST_DEVICE constexpr 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 Func > | |
void | ParallelFor (size_t size, size_t nthreads, Func fn) |
Variables | |
constexpr std::size_t | dynamic_extent = std::numeric_limits<std::size_t>::max() |
constexpr double | kPI = 3.14159265358979323846 |
Constant PI. More... | |
constexpr double | kEulerMascheroni = 0.57721566490153286060651209008240243104215933593992 |
The Euler-Mascheroni_constant. More... | |
constexpr size_t | kBlockThreads = 256 |
using xgboost::common::CompressedByteT = typedef unsigned char |
using xgboost::common::GHistIndexRow = typedef Span<uint32_t const> |
A single row in global histogram index. Directly represent the global index in the histogram entry.
using xgboost::common::GHistRow = typedef Span<xgboost::detail::GradientPairInternal<GradientSumT> > |
using xgboost::common::GlobalRandomEngine = typedef RandomEngine |
global random engine
using xgboost::common::MemoryBufferStream = typedef rabit::utils::MemoryBufferStream |
using xgboost::common::MemoryFixSizeBuffer = typedef rabit::utils::MemoryFixSizeBuffer |
using xgboost::common::RandomEngine = typedef std::mt19937 |
Define mt19937 as default type Random Engine.
|
strong |
|
strong |
|
strong |
int xgboost::common::AllVisibleGPUs | ( | ) |
std::vector<Idx> xgboost::common::ArgSort | ( | std::vector< V > const & | array, |
Comp | comp = std::less<V>{} |
||
) |
XGBOOST_DEVICE auto xgboost::common::as_bytes | ( | Span< T, E > | s | ) | -> Span<const byte, detail::ExtentAsBytesValue<T, E>::value> |
XGBOOST_DEVICE auto xgboost::common::as_writable_bytes | ( | Span< T, E > | s | ) | -> Span<byte, detail::ExtentAsBytesValue<T, E>::value> |
|
inline |
|
inline |
XGBOOST_DEVICE std::enable_if< std::numeric_limits<T>::is_integer, bool>::type xgboost::common::CheckNAN | ( | T | ) |
|
inline |
|
inline |
XGBOOST_DEVICE constexpr bool xgboost::common::CloseTo | ( | T | a, |
U | b | ||
) |
Equality test for both integer and floating point.
void xgboost::common::CopyHist | ( | GHistRow< GradientSumT > | dst, |
const GHistRow< GradientSumT > | src, | ||
size_t | begin, | ||
size_t | end | ||
) |
Copy hist from src to dst in range [begin, end)
XGBOOST_DEVICE T1 xgboost::common::DivRoundUp | ( | const T1 | a, |
const T2 | b | ||
) |
|
inline |
|
inline |
Find the maximum iterator within the iterators.
begin | The begining iterator. |
end | The end iterator. |
Iterator | The type of the iterator. |
GlobalRandomEngine& xgboost::common::GlobalRandom | ( | ) |
global singleton of a random engine. This random engine is thread-local and only visible to current thread.
void xgboost::common::IncrementHist | ( | GHistRow< GradientSumT > | dst, |
const GHistRow< GradientSumT > | add, | ||
size_t | begin, | ||
size_t | end | ||
) |
Increment hist as dst += add in range [begin, end)
void xgboost::common::InitilizeHistByZeroes | ( | GHistRow< GradientSumT > | hist, |
size_t | begin, | ||
size_t | end | ||
) |
fill a histogram by zeros
std::string xgboost::common::LoadSequentialFile | ( | std::string | uri, |
bool | stream = false |
||
) |
Helper function for loading consecutive file to avoid dmlc Stream when possible.
uri | URI or file name to file. |
stream | Use dmlc Stream unconditionally if set to true. Used for running test without remote filesystem. |
|
inline |
|
inline |
perform numerically safe logsum
x | left input operand |
y | right input operand |
|
inline |
perform numerically safe logsum
begin | The begining iterator. |
end | The end iterator. |
Iterator | The type of the iterator. |
XGBOOST_DEVICE T xgboost::common::Max | ( | T | a, |
T | b | ||
) |
XGBOOST_DEVICE constexpr bool xgboost::common::operator!= | ( | Span< T, X > | l, |
Span< U, Y > | r | ||
) |
XGBOOST_DEVICE constexpr bool xgboost::common::operator< | ( | Span< T, X > | l, |
Span< U, Y > | r | ||
) |
XGBOOST_DEVICE constexpr bool xgboost::common::operator<= | ( | Span< T, X > | l, |
Span< U, Y > | r | ||
) |
XGBOOST_DEVICE bool xgboost::common::operator== | ( | Span< T, X > | l, |
Span< U, Y > | r | ||
) |
XGBOOST_DEVICE constexpr bool xgboost::common::operator> | ( | Span< T, X > | l, |
Span< U, Y > | r | ||
) |
XGBOOST_DEVICE constexpr bool xgboost::common::operator>= | ( | Span< T, X > | l, |
Span< U, Y > | r | ||
) |
void xgboost::common::ParallelFor | ( | size_t | size, |
size_t | nthreads, | ||
Func | fn | ||
) |
void xgboost::common::ParallelFor2d | ( | const BlockedSpace2d & | space, |
int | nthreads, | ||
Func | func | ||
) |
|
inline |
calculate the sigmoid of the input.
x | input parameter |
|
inline |
|
inline |
Do inplace softmax transformaton on start to end.
Iterator | Input iterator type |
start | Start iterator of input |
end | end iterator of input |
|
inline |
Split a string by delimiter.
s | String to be splitted. |
delim | The delimiter. |
void xgboost::common::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)
|
inline |
std::vector<T> xgboost::common::WeightedSamplingWithoutReplacement | ( | std::vector< T > const & | array, |
std::vector< float > const & | weights, | ||
size_t | n | ||
) |
constexpr std::size_t xgboost::common::dynamic_extent = std::numeric_limits<std::size_t>::max() |
constexpr size_t xgboost::common::kBlockThreads = 256 |
constexpr double xgboost::common::kEulerMascheroni = 0.57721566490153286060651209008240243104215933593992 |
The Euler-Mascheroni_constant.
constexpr double xgboost::common::kPI = 3.14159265358979323846 |
Constant PI.