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

Fix metric name loading. (#5122)

Browse Source
This commit is contained in:
Jiaming Yuan 2019-12-16 10:14:02 +08:00 committed by GitHub
parent 3136185bc5
commit 2fdb34ed2e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/learner.cc
View File

@ -333,7 +333,7 @@ class LearnerImpl : public Learner {
for (size_t i = 0; i < n_metrics; ++i) { for (size_t i = 0; i < n_metrics; ++i) {
metric_names_[i]= get<String>(j_metrics[i]); metric_names_[i]= get<String>(j_metrics[i]);
metrics_[i] = std::unique_ptr<Metric>( metrics_[i] = std::unique_ptr<Metric>(
Metric::Create(metric_names_.back(), &generic_parameters_)); Metric::Create(metric_names_[i], &generic_parameters_));
} }
fromJson(learner_parameters.at("generic_param"), &generic_parameters_); fromJson(learner_parameters.at("generic_param"), &generic_parameters_);
@ -362,7 +362,7 @@ class LearnerImpl : public Learner {
for (size_t i = 0; i < metrics_.size(); ++i) { for (size_t i = 0; i < metrics_.size(); ++i) {
metrics[i] = String(metrics_[i]->Name()); metrics[i] = String(metrics_[i]->Name());
} }
learner_parameters["metrics"] = Array(metrics); learner_parameters["metrics"] = Array(std::move(metrics));
learner_parameters["generic_param"] = toJson(generic_parameters_); learner_parameters["generic_param"] = toJson(generic_parameters_);
} }