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

Merge pull request #1221 from ryaninhust/master

Browse Source 
[DATA] fix instance weights loading
This commit is contained in:
Tianqi Chen 2016-05-24 20:19:31 -07:00
commit 5c14daffe2
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;
}