Hendrik/xgboost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Avoid resetting seed for every configuration. (#6349)

Browse Source
This commit is contained in:
Jiaming Yuan
2020-11-06 10:28:35 +08:00
committed by GitHub
parent f3a4253984
commit 519cee115a
4 changed files with 41 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
include/xgboost/generic_parameters.h
View File

@@ -14,10 +14,11 @@ namespace xgboost {
struct GenericParameter : public XGBoostParameter<GenericParameter> {
// Constant representing the device ID of CPU.
static int32_t constexpr kCpuId = -1;
static int64_t constexpr kDefaultSeed = 0;
public:
// stored random seed
int64_t seed;
int64_t seed { kDefaultSeed };
// whether seed the PRNG each iteration
bool seed_per_iteration;
// number of threads to use if OpenMP is enabled
@@ -46,7 +47,7 @@ struct GenericParameter : public XGBoostParameter<GenericParameter> {
// declare parameters
DMLC_DECLARE_PARAMETER(GenericParameter) {
DMLC_DECLARE_FIELD(seed).set_default(0).describe(
DMLC_DECLARE_FIELD(seed).set_default(kDefaultSeed).describe(
"Random number seed during training.");
DMLC_DECLARE_ALIAS(seed, random_state);
DMLC_DECLARE_FIELD(seed_per_iteration)