Unverified Commit ed3d774c authored by Guolin Ke's avatar Guolin Ke Committed by GitHub
Browse files

fix max_drop checking

parent 04d37f9b
...@@ -454,7 +454,6 @@ void BoostingConfig::Set(const std::unordered_map<std::string, std::string>& par ...@@ -454,7 +454,6 @@ void BoostingConfig::Set(const std::unordered_map<std::string, std::string>& par
CHECK(drop_rate <= 1.0 && drop_rate >= 0.0); CHECK(drop_rate <= 1.0 && drop_rate >= 0.0);
CHECK(skip_drop <= 1.0 && skip_drop >= 0.0); CHECK(skip_drop <= 1.0 && skip_drop >= 0.0);
GetInt(params, "max_drop", &max_drop); GetInt(params, "max_drop", &max_drop);
CHECK(max_drop > 0);
GetBool(params, "xgboost_dart_mode", &xgboost_dart_mode); GetBool(params, "xgboost_dart_mode", &xgboost_dart_mode);
GetBool(params, "uniform_drop", &uniform_drop); GetBool(params, "uniform_drop", &uniform_drop);
GetDouble(params, "top_rate", &top_rate); GetDouble(params, "top_rate", &top_rate);
......
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