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

Fix issue #1236: cli_main crashes when dumping count:poisson model (#1253)

Browse Source
This commit is contained in:
Shengwen Yang 2016-06-08 12:52:47 +08:00 committed by Tianqi Chen
parent e2c1aa8b51
commit e034fdf74c
2 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
dmlc-core

@ -1 +1 @@
Subproject commit 1db0792e1a55355b1f07699bba18c88ded996953 Subproject commit 9fd3b48462a7a651e12a197679f71e043dcb25a2

1
src/cli_main.cc
View File

@ -271,6 +271,7 @@ void CLIDump2Text(const CLIParam& param) {
std::unique_ptr<Learner> learner(Learner::Create({})); std::unique_ptr<Learner> learner(Learner::Create({}));
std::unique_ptr<dmlc::Stream> fi( std::unique_ptr<dmlc::Stream> fi(
dmlc::Stream::Create(param.model_in.c_str(), "r")); dmlc::Stream::Create(param.model_in.c_str(), "r"));
learner->Configure(param.cfg);
learner->Load(fi.get()); learner->Load(fi.get());
// dump data // dump data
std::vector<std::string> dump = learner->Dump2Text(fmap, param.dump_stats); std::vector<std::string> dump = learner->Dump2Text(fmap, param.dump_stats);