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

adding a copy of base_margin to slice, fixes a bug where base_margin was notcopied during cross-validation (#2007)

Browse Source
This commit is contained in:
Oleg Sofrygin 2017-03-16 10:36:57 -07:00 committed by Tianqi Chen
parent 3a2b8332a6
commit 9d19e13ed0
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
src/c_api/c_api.cc
View File

@ -413,6 +413,9 @@ XGB_DLL int XGDMatrixSliceDMatrix(DMatrixHandle handle,
if (src.info.weights.size() != 0) {
ret.info.weights.push_back(src.info.weights[ridx]);
}
if (src.info.base_margin.size() != 0) {
ret.info.base_margin.push_back(src.info.base_margin[ridx]);
}
if (src.info.root_index.size() != 0) {
ret.info.root_index.push_back(src.info.root_index[ridx]);
}