|
xgboost
|
用于构建直方图索引批次的参数。 更多...
#include <data.h>

公共成员函数 | |
| 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} |
| 为外部内存预取批次的数量。更多... | |
用于构造直方图索引批次的参数。
|
默认 |
精确或其他不需要直方图的方法。
|
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,用于未来近似实现中的草图。
| 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()} |
用于生成直方图列矩阵的参数。