fix cli_main crashes when using count:poisson regression

2016-05-26 10:03:29 -07:00 · 2016-05-26 10:03:29 -07:00 · 1dde863c98
commit 1dde863c98
parent 2ef81e0673
1 changed files with 1 additions and 0 deletions
--- a/src/cli_main.cc
+++ b/src/cli_main.cc
@ -297,6 +297,7 @@ void CLIPredict(const CLIParam& param) {
  std::unique_ptr<Learner> learner(Learner::Create({}));
  std::unique_ptr<dmlc::Stream> fi(
      dmlc::Stream::Create(param.model_in.c_str(), "r"));
+  learner->Configure(param.cfg);
  learner->Load(fi.get());

  if (param.silent == 0) {