From 1dde863c98a40fbda4188f89892f0c8acce3c45c Mon Sep 17 00:00:00 2001 From: Zhongliang Li Date: Thu, 26 May 2016 10:03:29 -0700 Subject: [PATCH] fix cli_main crashes when using count:poisson regression --- src/cli_main.cc | 1 + 1 file changed, 1 insertion(+) diff --git a/src/cli_main.cc b/src/cli_main.cc index 039ce4070..f85202aef 100644 --- a/src/cli_main.cc +++ b/src/cli_main.cc @@ -297,6 +297,7 @@ void CLIPredict(const CLIParam& param) { std::unique_ptr learner(Learner::Create({})); std::unique_ptr fi( dmlc::Stream::Create(param.model_in.c_str(), "r")); + learner->Configure(param.cfg); learner->Load(fi.get()); if (param.silent == 0) {