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
Download Patch File Download Diff File Expand all files Collapse all files

2
dmlc-core

Submodule dmlc-core updated: 1db0792e1a...9fd3b48462

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);