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(); pcol->Clear();
utils::ParallelGroupBuilder<SparseBatch::Entry> utils::ParallelGroupBuilder<SparseBatch::Entry>
builder(&pcol->offset, &pcol->data); builder(&pcol->offset, &pcol->data);
builder.InitBudget(0, nthread); builder.InitBudget(info_.num_col(), nthread);
// start working // start working
iter_->BeforeFirst(); iter_->BeforeFirst();
while (iter_->Next()) { 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 // sort columns
bst_omp_uint ncol = static_cast<bst_omp_uint>(pcol->Size()); bst_omp_uint ncol = static_cast<bst_omp_uint>(pcol->Size());
#pragma omp parallel for schedule(dynamic, 1) num_threads(nthread) #pragma omp parallel for schedule(dynamic, 1) num_threads(nthread)