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

[DATA] fix instance weights loading

Browse Source
This commit is contained in:
yuanbowen 2016-05-23 18:40:41 +08:00
parent c85b9012c6
commit 5898f1c59e
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/data/data.cc
View File

@ -223,6 +223,10 @@ DMatrix* DMatrix::Load(const std::string& uri,
LOG(CONSOLE) << info.base_margin.size()
<< " base_margin are loaded from " << fname << ".base_margin";
}
if (MetaTryLoadFloatInfo(fname + ".weight", &info.weights) && !silent) {
LOG(CONSOLE) << info.weights.size()
<< " weights are loaded from " << fname << ".weight";
}
}
return dmat;
}