xgboost
公共成员函数 | 公共成员属性 | 所有成员列表
xgboost::BatchParam 结构体参考

用于构建直方图索引批次的参数。 更多...

#include <data.h>

xgboost::BatchParam 的协作图
Collaboration graph

公共成员函数

 BatchParam ()=default
 Exact 或其他不需要直方图的方法。 更多...
 
 BatchParam (bst_bin_t max_bin, double sparse_thresh)
 供 hist 树方法使用。 更多...
 
 BatchParam (bst_bin_t max_bin, common::Span< float const > hessian, bool regenerate)
 供 approx 树方法使用。 更多...
 
bool ParamNotEqual (BatchParam const &other) const
 
bool Initialized () const
 
BatchParam MakeCache () const
 DMatrix 创建自身的副本,以描述其现有索引是如何生成的。 更多...
 

公共属性

bst_bin_t max_bin {0}
 每个特征用于直方图的最大 bin 数。 更多...
 
common::Span< float const > hess
 Hessian,用于未来 approx 实现的草图绘制。 更多...
 
bool regen {false}
 是否应强制 DMatrix 重新生成批次。仅用于 GHistIndex。 更多...
 
bool forbid_regen {false}
 禁止重新生成梯度索引。用于内部验证。 更多...
 
double sparse_thresh {std::numeric_limits<double>::quiet_NaN()}
 用于为 hist 生成列矩阵的参数。 更多...
 
bool prefetch_copy {true}
 用于 GPU 外部内存。是否将数据复制到设备。 更多...
 
std::int32_t n_prefetch_batches {3}
 为外部内存预取的批次数。 更多...
 

详细描述

用于构建直方图索引批次的参数。

构造函数与析构函数文档

◆ BatchParam() [1/3]

xgboost::BatchParam::BatchParam ( )
default

Exact 或其他不需要直方图的方法。

◆ BatchParam() [2/3]

xgboost::BatchParam::BatchParam ( bst_bin_t  max_bin,
double  sparse_thresh 
)
inline

供 hist 树方法使用。

◆ BatchParam() [3/3]

xgboost::BatchParam::BatchParam ( bst_bin_t  max_bin,
common::Span< float const >  hessian,
bool  regenerate 
)
inline

供 approx 树方法使用。

获取按 hessian 加权的草图批次。如果 span 发生变化,批次将重新生成,因此调用者应在每次迭代中保留该 span。

成员函数文档

◆ Initialized()

bool xgboost::BatchParam::Initialized ( ) const
inline

◆ MakeCache()

BatchParam xgboost::BatchParam::MakeCache ( ) const
inline

DMatrix 创建自身的副本,以描述其现有索引是如何生成的。

◆ ParamNotEqual()

bool xgboost::BatchParam::ParamNotEqual ( BatchParam const &  other) const
inline

成员数据文档

◆ forbid_regen

bool xgboost::BatchParam::forbid_regen {false}

禁止重新生成梯度索引。用于内部验证。

◆ hess

common::Span<float const> xgboost::BatchParam::hess

Hessian,用于未来 approx 实现的草图绘制。

◆ max_bin

bst_bin_t xgboost::BatchParam::max_bin {0}

每个特征用于直方图的最大 bin 数。

◆ n_prefetch_batches

std::int32_t xgboost::BatchParam::n_prefetch_batches {3}

为外部内存预取的批次数。

◆ prefetch_copy

bool xgboost::BatchParam::prefetch_copy {true}

用于 GPU 外部内存。是否将数据复制到设备。

这仅影响当前的迭代轮次。

◆ regen

bool xgboost::BatchParam::regen {false}

是否应强制 DMatrix 重新生成批次。仅用于 GHistIndex。

◆ sparse_thresh

double xgboost::BatchParam::sparse_thresh {std::numeric_limits<double>::quiet_NaN()}

用于为 hist 生成列矩阵的参数。


此结构体的文档由以下文件生成