Commit 9f04f276 authored by Guolin Ke's avatar Guolin Ke
Browse files

change dropping_seed to drop_seed

parent 47b7455d
...@@ -192,7 +192,7 @@ public: ...@@ -192,7 +192,7 @@ public:
int early_stopping_round = 0; int early_stopping_round = 0;
int num_class = 1; int num_class = 1;
double drop_rate = 0.01; double drop_rate = 0.01;
int dropping_seed = 4; int drop_seed = 4;
void Set(const std::unordered_map<std::string, std::string>& params) override; void Set(const std::unordered_map<std::string, std::string>& params) override;
}; };
......
...@@ -29,7 +29,7 @@ void DART::Init(const BoostingConfig* config, const Dataset* train_data, const O ...@@ -29,7 +29,7 @@ void DART::Init(const BoostingConfig* config, const Dataset* train_data, const O
gbdt_config_ = dynamic_cast<const GBDTConfig*>(config); gbdt_config_ = dynamic_cast<const GBDTConfig*>(config);
drop_rate_ = gbdt_config_->drop_rate; drop_rate_ = gbdt_config_->drop_rate;
shrinkage_rate_ = 1.0; shrinkage_rate_ = 1.0;
random_for_drop_ = Random(gbdt_config_->dropping_seed); random_for_drop_ = Random(gbdt_config_->drop_seed);
} }
bool DART::TrainOneIter(const score_t* gradient, const score_t* hessian, bool is_eval) { bool DART::TrainOneIter(const score_t* gradient, const score_t* hessian, bool is_eval) {
......
...@@ -298,7 +298,7 @@ void BoostingConfig::Set(const std::unordered_map<std::string, std::string>& par ...@@ -298,7 +298,7 @@ void BoostingConfig::Set(const std::unordered_map<std::string, std::string>& par
CHECK(output_freq >= 0); CHECK(output_freq >= 0);
GetBool(params, "is_training_metric", &is_provide_training_metric); GetBool(params, "is_training_metric", &is_provide_training_metric);
GetInt(params, "num_class", &num_class); GetInt(params, "num_class", &num_class);
GetInt(params, "dropping_seed", &dropping_seed); GetInt(params, "drop_seed", &drop_seed);
GetDouble(params, "drop_rate", &drop_rate); GetDouble(params, "drop_rate", &drop_rate);
CHECK(drop_rate <= 1.0 && drop_rate >= 0.0); CHECK(drop_rate <= 1.0 && drop_rate >= 0.0);
} }
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment