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

fix col from dense

Browse Source
This commit is contained in:
tqchen 2015-06-15 09:24:10 -07:00
parent c51d71b033
commit 7a92d4008e
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
src/io/simple_fmatrix-inl.hpp
View File

@ -153,7 +153,7 @@ class FMatrixS : public IFMatrix {
pcol->Clear();
utils::ParallelGroupBuilder<SparseBatch::Entry>
builder(&pcol->offset, &pcol->data);
builder.InitBudget(0, nthread);
builder.InitBudget(info_.num_col(), nthread);
// start working
iter_->BeforeFirst();
while (iter_->Next()) {
@ -204,7 +204,8 @@ class FMatrixS : public IFMatrix {
}
}
utils::Assert(pcol->Size() == info_.num_col(), "inconsistent col data");
utils::Assert(pcol->Size() == info_.num_col(),
"inconsistent col data");
// sort columns
bst_omp_uint ncol = static_cast<bst_omp_uint>(pcol->Size());
#pragma omp parallel for schedule(dynamic, 1) num_threads(nthread)