From 2fdb34ed2ea5fbe4d690332d4658cc34260c0edc Mon Sep 17 00:00:00 2001 From: Jiaming Yuan Date: Mon, 16 Dec 2019 10:14:02 +0800 Subject: [PATCH] Fix metric name loading. (#5122) --- src/learner.cc | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/learner.cc b/src/learner.cc index f6f200774..45d3efff1 100644 --- a/src/learner.cc +++ b/src/learner.cc @@ -333,7 +333,7 @@ class LearnerImpl : public Learner { for (size_t i = 0; i < n_metrics; ++i) { metric_names_[i]= get(j_metrics[i]); metrics_[i] = std::unique_ptr( - 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_); }