xgboost
|
Parameters for constructing batches. More...
#include <data.h>
Public Member Functions | |
BatchParam ()=default | |
BatchParam (int32_t device, int32_t max_bin) | |
BatchParam (int32_t device, int32_t max_bin, common::Span< float > hessian, bool regenerate=false) | |
Get batch with sketch weighted by hessian. The batch will be regenerated if the span is changed, so caller should keep the span for each iteration. More... | |
bool | operator!= (const BatchParam &other) const |
Public Attributes | |
int | gpu_id {-1} |
The GPU device to use. More... | |
int | max_bin {0} |
Maximum number of bins per feature for histograms. More... | |
common::Span< float > | hess |
Hessian, used for sketching with future approx implementation. More... | |
bool | regen {false} |
Whether should DMatrix regenerate the batch. Only used for GHistIndex. More... | |
Parameters for constructing batches.
|
default |
|
inline |
|
inline |
Get batch with sketch weighted by hessian. The batch will be regenerated if the span is changed, so caller should keep the span for each iteration.
|
inline |
int xgboost::BatchParam::gpu_id {-1} |
The GPU device to use.
common::Span<float> xgboost::BatchParam::hess |
Hessian, used for sketching with future approx implementation.
int xgboost::BatchParam::max_bin {0} |
Maximum number of bins per feature for histograms.
bool xgboost::BatchParam::regen {false} |
Whether should DMatrix regenerate the batch. Only used for GHistIndex.