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

Merge pull request #1232 from zl1zl/master

Browse Source 
fix cli_main crashes when using count:poisson regression
This commit is contained in:
Tianqi Chen 2016-05-27 20:22:50 -07:00
commit 6e3463097d
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
src/cli_main.cc
View File

@ -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) {