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:
int early_stopping_round = 0;
int num_class = 1;
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;
};
......
......@@ -29,7 +29,7 @@ void DART::Init(const BoostingConfig* config, const Dataset* train_data, const O
gbdt_config_ = dynamic_cast<const GBDTConfig*>(config);
drop_rate_ = gbdt_config_->drop_rate;
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) {
......
......@@ -298,7 +298,7 @@ void BoostingConfig::Set(const std::unordered_map<std::string, std::string>& par
CHECK(output_freq >= 0);
GetBool(params, "is_training_metric", &is_provide_training_metric);
GetInt(params, "num_class", &num_class);
GetInt(params, "dropping_seed", &dropping_seed);
GetInt(params, "drop_seed", &drop_seed);
GetDouble(params, "drop_rate", &drop_rate);
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