Commit fe8e2162 authored by Guolin Ke's avatar Guolin Ke
Browse files

fix `output_freq` alias.

parent 487fbbc9
...@@ -444,6 +444,7 @@ struct ParameterAlias { ...@@ -444,6 +444,7 @@ struct ParameterAlias {
{ "workers", "machines" }, { "workers", "machines" },
{ "nodes", "machines" }, { "nodes", "machines" },
{ "subsample_for_bin", "bin_construct_sample_cnt" }, { "subsample_for_bin", "bin_construct_sample_cnt" },
{ "metric_freq", "output_freq" }
}); });
const std::unordered_set<std::string> parameter_set({ const std::unordered_set<std::string> parameter_set({
"config", "config_file", "task", "device", "config", "config_file", "task", "device",
......
...@@ -422,7 +422,7 @@ void BoostingConfig::Set(const std::unordered_map<std::string, std::string>& par ...@@ -422,7 +422,7 @@ void BoostingConfig::Set(const std::unordered_map<std::string, std::string>& par
CHECK(learning_rate > 0.0f); CHECK(learning_rate > 0.0f);
GetInt(params, "early_stopping_round", &early_stopping_round); GetInt(params, "early_stopping_round", &early_stopping_round);
CHECK(early_stopping_round >= 0); CHECK(early_stopping_round >= 0);
GetInt(params, "metric_freq", &output_freq); GetInt(params, "output_freq", &output_freq);
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);
......
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