xgboost
|
Parameters for constructing batches. More...
#include <data.h>
Public Member Functions | |
BatchParam ()=default | |
BatchParam (int32_t device, int32_t max_bin, size_t gpu_page_size=0) | |
bool | operator!= (const BatchParam &other) const |
Public Attributes | |
int | gpu_id |
The GPU device to use. More... | |
int | max_bin {0} |
Maximum number of bins per feature for histograms. More... | |
size_t | gpu_page_size |
Page size for external memory mode. More... | |
Parameters for constructing batches.
|
default |
|
inline |
|
inline |
int xgboost::BatchParam::gpu_id |
The GPU device to use.
size_t xgboost::BatchParam::gpu_page_size |
Page size for external memory mode.
int xgboost::BatchParam::max_bin {0} |
Maximum number of bins per feature for histograms.