xgboost
| 命名空间 | | 类型定义 | 变量
base.h 文件参考

定义 xgboost 的配置宏和基本类型。更多...

#include <dmlc/omp.h>
#include <xgboost/windefs.h>
#include <cstdint>
#include <ostream>
#include <string>
#include <utility>
#include <vector>
base.h 的包含依赖图
此图显示哪些文件直接或间接包含此文件

转到此文件的源代码。

类  xgboost::detail::GradientPairInternal< T >
 梯度统计对的实现。模板特化可用于重载不同的梯度类型,例如低精度、高精度、整数、浮点数。更多...
 
类  xgboost::GradientPairInt64
 用于高精度梯度对的定点表示。具有不同的接口,因此我们不会在增益计算中意外使用它。更多...
 

命名空间

 xgboost
 Learner 接口,将目标、gbm 和评估整合在一起。这是用户界面 XGBoost 训练模块。
 
 xgboost::detail
 

#define XGBOOST_STRICT_R_MODE   0
 R 库的字符串标志,用于在需要时留下钩子。更多...
 
#define XGBOOST_LOG_WITH_TIME   1
 是否始终记录带时间的控制台消息。它将显示为,时间戳附加到消息的开头。"[21:47:50] 6513x126 matrix with 143286 entries loaded from ../data/agaricus.txt.train"。更多...
 
#define XGBOOST_CUSTOMIZE_GLOBAL_PRNG   0
 是否自定义全局伪随机数生成器(PRNG)。更多...
 
#define XGBOOST_ALIGNAS(X)
 检查是否支持 alignas(*) 关键字。(g++ 4.8 或更高版本)更多...
 
#define XGBOOST_EXPECT(cond, ret)   (cond)
 
#define XGBOOST_DEVICE
 标记函数可用于设备。更多...
 
#define XGBOOST_HOST_DEV_INLINE
 
#define XGBOOST_DEV_INLINE
 
#define XGBOOST_RESTRICT   __restrict__
 

类型定义

使用 xgboost::bst_uint = std::uint32_t
 用于特征索引的无符号整数类型。更多...
 
使用 xgboost::bst_ulong = std::uint64_t
 无符号长整数。更多...
 
使用 xgboost::bst_float = float
 浮点类型,用于存储统计数据。更多...
 
使用 xgboost::bst_cat_t = std::int32_t
 分类值类型。更多...
 
使用 xgboost::bst_feature_t = std::uint32_t
 数据列(特征)索引的类型。更多...
 
使用 xgboost::bst_bin_t = std::int32_t
 直方图 bin 索引的类型。我们有时使用 -1 表示无效 bin。更多...
 
使用 xgboost::bst_idx_t = std::uint64_t
 数据行索引(样本)的类型。更多...
 
使用 xgboost::bst_node_t = std::int32_t
 树节点索引的类型。更多...
 
使用 xgboost::bst_group_t = std::uint32_t
 排名组索引的类型。更多...
 
使用 xgboost::bst_target_t = std::uint32_t
 用于索引输出目标的类型。更多...
 
使用 xgboost::bst_layer_t = std::int32_t
 用于索引增强层(boosted layers)的类型。更多...
 
使用 xgboost::bst_tree_t = std::int32_t
 用于索引树的类型。更多...
 
使用 xgboost::bst_d_ordinal_t = std::int16_t
 CUDA 设备的序号。更多...
 
使用 xgboost::GradientPair = detail::GradientPairInternal< float >
 梯度增强中通常需要的梯度统计对。更多...
 
使用 xgboost::GradientPairPrecise = detail::GradientPairInternal< double >
 高精度梯度统计对。更多...
 
使用 xgboost::Args = std::vector< std::pair< std::string, std::string > >
 
使用 xgboost::omp_ulong = dmlc::omp_ulong
 为 openmp 循环定义无符号长整数。更多...
 
使用 xgboost::bst_omp_uint = dmlc::omp_uint
 为 openmp 循环定义无符号整数。更多...
 
使用 xgboost::XGBoostVersionT = std::int32_t
 用于以二进制形式表示版本号的类型。更多...
 

变量

constexpr bst_float xgboost::kRtEps = 1e-6f
 用于最小分裂决策的小 eps 间隙。更多...
 

详细描述

为 xgboost 定义配置宏和基本类型。

版权所有 2015-2025, XGBoost 贡献者

宏定义文档

◆ XGBOOST_ALIGNAS

#define XGBOOST_ALIGNAS (   X)

检查是否支持 alignas(*) 关键字。(g++ 4.8 或更高版本)

◆ XGBOOST_CUSTOMIZE_GLOBAL_PRNG

#define XGBOOST_CUSTOMIZE_GLOBAL_PRNG   0

是否自定义全局伪随机数生成器(PRNG)。

◆ XGBOOST_DEV_INLINE

#define XGBOOST_DEV_INLINE

◆ XGBOOST_DEVICE

#define XGBOOST_DEVICE

Tag function as usable by device.

◆ XGBOOST_EXPECT

#define XGBOOST_EXPECT (   cond,
  ret 
)    (cond)

◆ XGBOOST_HOST_DEV_INLINE

#define XGBOOST_HOST_DEV_INLINE

◆ XGBOOST_LOG_WITH_TIME

#define XGBOOST_LOG_WITH_TIME   1

是否始终记录带时间的控制台消息。它将显示为,时间戳附加到消息的开头。"[21:47:50] 6513x126 matrix with 143286 entries loaded from ../data/agaricus.txt.train"。

◆ XGBOOST_RESTRICT

#define XGBOOST_RESTRICT   __restrict__

◆ XGBOOST_STRICT_R_MODE

#define XGBOOST_STRICT_R_MODE   0

R 库的字符串标志,用于在需要时留下钩子。