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

Export JSON config in get_params. (#5256)

Browse Source
This commit is contained in:
Jiaming Yuan
2020-02-03 12:46:51 +08:00
committed by Hyunsu Philip Cho
parent e7e522fb06
commit ed37fdb9c9
3 changed files with 56 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
src/learner.cc
View File

@@ -269,10 +269,7 @@ class LearnerImpl : public Learner {
}
}
}
auto learner_model_param = mparam_.ToJson();
for (auto const& kv : get<Object>(learner_model_param)) {
keys.emplace_back(kv.first);
}
keys.emplace_back(kEvalMetric);
keys.emplace_back("verbosity");
keys.emplace_back("num_output_group");
@@ -425,6 +422,7 @@ class LearnerImpl : public Learner {
auto& learner_parameters = out["learner"];
learner_parameters["learner_train_param"] = toJson(tparam_);
learner_parameters["learner_model_param"] = mparam_.ToJson();
learner_parameters["gradient_booster"] = Object();
auto& gradient_booster = learner_parameters["gradient_booster"];
gbm_->SaveConfig(&gradient_booster);