From e034fdf74cadeab27365d3152461c6d6c6431dd1 Mon Sep 17 00:00:00 2001 From: Shengwen Yang Date: Wed, 8 Jun 2016 12:52:47 +0800 Subject: [PATCH] Fix issue #1236: cli_main crashes when dumping count:poisson model (#1253) --- dmlc-core | 2 +- src/cli_main.cc | 1 + 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/dmlc-core b/dmlc-core index 1db0792e1..9fd3b4846 160000 --- a/dmlc-core +++ b/dmlc-core @@ -1 +1 @@ -Subproject commit 1db0792e1a55355b1f07699bba18c88ded996953 +Subproject commit 9fd3b48462a7a651e12a197679f71e043dcb25a2 diff --git a/src/cli_main.cc b/src/cli_main.cc index f85202aef..e79592615 100644 --- a/src/cli_main.cc +++ b/src/cli_main.cc @@ -271,6 +271,7 @@ void CLIDump2Text(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()); // dump data std::vector dump = learner->Dump2Text(fmap, param.dump_stats);