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

Correct mistake from dmatrix refactor (#3408)

Browse Source
This commit is contained in:
Rory Mitchell
2018-07-24 15:03:36 +12:00
committed by GitHub
parent e9a97e0d88
commit a725272e19
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
src/tree/updater_histmaker.cc
View File

@@ -434,9 +434,10 @@ class CQHistMaker: public HistMaker<TStats> {
this->CorrectNonDefaultPositionByBatch(batch, fsplit_set_, tree);
// start enumeration
const auto nsize = static_cast<bst_omp_uint>(batch.Size());
const auto nsize = static_cast<bst_omp_uint>(work_set_.size());
#pragma omp parallel for schedule(dynamic, 1)
for (bst_omp_uint fid = 0; fid < nsize; ++fid) {
for (bst_omp_uint i = 0; i < nsize; ++i) {
int fid = work_set_[i];
int offset = feat2workindex_[fid];
if (offset >= 0) {
this->UpdateSketchCol(gpair, batch[fid], tree,