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) {
metric_names_[i]= get<String>(j_metrics[i]);
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_);
@ -362,7 +362,7 @@ class LearnerImpl : public Learner {
for (size_t i = 0; i < metrics_.size(); ++i) {
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_);
}