Fix model parameter recovery (#4738)

2019-08-07 14:32:10 +08:00
parent 851b5b3808
commit 3e2c472944
2 changed files with 43 additions and 0 deletions
--- a/src/learner.cc
+++ b/src/learner.cc
@@ -287,9 +287,13 @@ class LearnerImpl : public Learner {
        metrics_.emplace_back(Metric::Create(name, &generic_param_));
      }
    }
+
    cfg_["num_class"] = common::ToString(mparam_.num_class);
    cfg_["num_feature"] = common::ToString(mparam_.num_feature);

+    auto n = tparam_.__DICT__();
+    cfg_.insert(n.cbegin(), n.cend());
+
    gbm_->Configure({cfg_.cbegin(), cfg_.cend()});
    obj_->Configure({cfg_.begin(), cfg_.end()});