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 max_bin, double sparse_thresh) | |
BatchParam (int32_t max_bin, common::Span< float > hessian, bool regenerate) | |
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!= (BatchParam const &other) const |
bool | operator== (BatchParam const &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... | |
double | sparse_thresh {std::numeric_limits<double>::quiet_NaN()} |
Parameter used to generate column matrix for hist. More... | |
Parameters for constructing batches.
|
default |
|
inline |
|
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 |
|
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.
double xgboost::BatchParam::sparse_thresh {std::numeric_limits<double>::quiet_NaN()} |
Parameter used to generate column matrix for hist.