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

fix bug in loading config for pred task (#2795)

Browse Source
This commit is contained in:
Qiang Luo 2017-10-20 13:10:14 +08:00 committed by Vadim Khotilovich
parent 5dca6745e1
commit c09ad421a8
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/cli_main.cc
View File

@ -318,8 +318,8 @@ 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());
learner->Configure(param.cfg);
if (param.silent == 0) {
LOG(CONSOLE) << "start prediction...";