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

safe fix

Browse Source
This commit is contained in:
tqchen 2015-04-06 14:53:40 -07:00
parent 99f8dd280e
commit 01771c813d
1 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/learner/learner-inl.hpp
View File

@ -173,9 +173,11 @@ class BoostLearner : public rabit::ISerializable {
utils::Check(fi.Read(&gap, sizeof(gap)) != 0, "BoostLearner: wrong model format"); utils::Check(fi.Read(&gap, sizeof(gap)) != 0, "BoostLearner: wrong model format");
len = len >> 32UL; len = len >> 32UL;
} }
if (len != 0) {
name_obj_.resize(len); name_obj_.resize(len);
utils::Check(fi.Read(&name_obj_[0], len) != 0, "BoostLearner: wrong model format"); utils::Check(fi.Read(&name_obj_[0], len) != 0, "BoostLearner: wrong model format");
} }
}
utils::Check(fi.Read(&name_gbm_), "BoostLearner: wrong model format"); utils::Check(fi.Read(&name_gbm_), "BoostLearner: wrong model format");
// delete existing gbm if any // delete existing gbm if any
if (obj_ != NULL) delete obj_; if (obj_ != NULL) delete obj_;