xgboost
|
用于构建直方图索引批次的参数。 更多...
#include <data.h>
公共成员函数 | |
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} |
为外部内存预取的批次数。 更多... | |
用于构建直方图索引批次的参数。
|
default |
Exact 或其他不需要直方图的方法。
|
inline |
供 hist 树方法使用。
|
inline |
供 approx 树方法使用。
获取按 hessian 加权的草图批次。如果 span 发生变化,批次将重新生成,因此调用者应在每次迭代中保留该 span。
|
inline |
|
inline |
为 DMatrix 创建自身的副本,以描述其现有索引是如何生成的。
|
inline |
bool xgboost::BatchParam::forbid_regen {false} |
禁止重新生成梯度索引。用于内部验证。
common::Span<float const> xgboost::BatchParam::hess |
Hessian,用于未来 approx 实现的草图绘制。
bst_bin_t xgboost::BatchParam::max_bin {0} |
每个特征用于直方图的最大 bin 数。
std::int32_t xgboost::BatchParam::n_prefetch_batches {3} |
为外部内存预取的批次数。
bool xgboost::BatchParam::prefetch_copy {true} |
用于 GPU 外部内存。是否将数据复制到设备。
这仅影响当前的迭代轮次。
bool xgboost::BatchParam::regen {false} |
是否应强制 DMatrix 重新生成批次。仅用于 GHistIndex。
double xgboost::BatchParam::sparse_thresh {std::numeric_limits<double>::quiet_NaN()} |
用于为 hist 生成列矩阵的参数。