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

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

#include <data.h>

xgboost::BatchParam 的协作图
Collaboration graph

公共成员函数

 BatchParam ()=default
 精确或其他不需要直方图的方法。更多...
 
 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,用于未来近似实现中的草图。更多...
 
bool regen {false}
 是否强制 DMatrix 重新生成批次。仅用于 GHistIndex。更多...
 
bool forbid_regen {false}
 禁止重新生成梯度索引。用于内部验证。更多...
 
double sparse_thresh {std::numeric_limits<double>::quiet_NaN()}
 用于生成直方图列矩阵的参数。更多...
 
bool prefetch_copy {true}
 用于 GPU 外部内存。是否将数据复制到设备。更多...
 
std::int32_t n_prefetch_batches {3}
 为外部内存预取批次的数量。更多...
 

详细描述

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

构造函数和析构函数文档

◆ BatchParam() [1/3]

xgboost::BatchParam::BatchParam ( )
默认

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

◆ 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 &  其他) const
inline

成员数据文档

◆ forbid_regen

bool xgboost::BatchParam::forbid_regen {false}

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

◆ hess

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

Hessian,用于未来近似实现中的草图。

◆ 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()}

用于生成直方图列矩阵的参数。


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