5 #ifndef XGBOOST_GENERIC_PARAMETERS_H_ 6 #define XGBOOST_GENERIC_PARAMETERS_H_ 8 #include <xgboost/logging.h> 16 static int32_t constexpr
kCpuId = -1;
34 if (this->n_gpus != 0) {
37 << this->__MANAGER__()->Find(
"n_gpus")->GetFieldInfo().description;
49 DMLC_DECLARE_FIELD(seed).set_default(0).describe(
50 "Random number seed during training.");
51 DMLC_DECLARE_ALIAS(seed, random_state);
52 DMLC_DECLARE_FIELD(seed_per_iteration)
55 "Seed PRNG determnisticly via iterator number, " 56 "this option will be switched on automatically on distributed " 58 DMLC_DECLARE_FIELD(nthread).set_default(0).describe(
59 "Number of threads to use.");
60 DMLC_DECLARE_ALIAS(nthread, n_jobs);
62 DMLC_DECLARE_FIELD(gpu_id)
65 .describe(
"The primary GPU device ordinal.");
66 DMLC_DECLARE_FIELD(gpu_page_size)
69 .describe(
"GPU page size when running in external memory mode.");
72 .describe(
"Enable using JSON for memory serialization (Python Pickle, " 73 "rabit checkpoints etc.).");
76 .describe(
"Enable checking whether parameters are used or not.");
77 DMLC_DECLARE_FIELD(n_gpus)
81 "\n\tDeprecated. Single process multi-GPU training is no longer supported." 82 "\n\tPlease switch to distributed training with one process per GPU." 83 "\n\tThis can be done using Dask or Spark. See documentation for details.");
92 #endif // XGBOOST_GENERIC_PARAMETERS_H_ int gpu_id
Definition: generic_parameters.h:27
int64_t seed
Definition: generic_parameters.h:20
static int32_t constexpr kCpuId
Definition: generic_parameters.h:16
Definition: parameter.h:84
Definition: generic_parameters.h:14
void CheckDeprecated()
Definition: generic_parameters.h:33
bool enable_experimental_json_serialization
Definition: generic_parameters.h:30
void ConfigureGpuId(bool require_gpu)
Configure the parameter `gpu_id'.
DMLC_DECLARE_PARAMETER(GenericParameter)
Definition: generic_parameters.h:48
namespace of xgboost
Definition: base.h:102
size_t gpu_page_size
Definition: generic_parameters.h:29
bool validate_parameters
Definition: generic_parameters.h:31
macro for using C++11 enum class as DMLC parameter
bool seed_per_iteration
Definition: generic_parameters.h:22
int nthread
Definition: generic_parameters.h:25