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

[TREE] switch to two pass

Browse Source
This commit is contained in:
tqchen 2016-01-19 21:55:33 -08:00
parent 523afcbcd2
commit a500fbc9b0
1 changed files with 10 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
src/tree/updater_histmaker.cc
View File

@ -361,8 +361,7 @@ class CQHistMaker: public HistMaker<TStats> {
}
void ResetPositionAfterSplit(DMatrix *p_fmat,
const RegTree &tree) override {
// remove this reset and do two pass reset on ResetPosAndPropose
this->ResetPositionCol(this->qexpand, p_fmat, tree);
this->GetSplitSet(this->qexpand, tree, &fsplit_set);
}
void ResetPosAndPropose(const std::vector<bst_gpair> &gpair,
DMatrix *p_fmat,
@ -398,11 +397,20 @@ class CQHistMaker: public HistMaker<TStats> {
// get smmary
thread_sketch.resize(this->get_nthread());
// TWOPASS: use the real set + split set in the column iteration.
this->SetDefaultPostion(p_fmat, tree);
work_set.insert(work_set.end(), fsplit_set.begin(), fsplit_set.end());
std::sort(work_set.begin(), work_set.end());
work_set.resize(std::unique(work_set.begin(), work_set.end()) - work_set.begin());
// start accumulating statistics
dmlc::DataIter<ColBatch> *iter = p_fmat->ColIterator(work_set);
iter->BeforeFirst();
while (iter->Next()) {
const ColBatch &batch = iter->Value();
// TWOPASS: use the real set + split set in the column iteration.
this->CorrectNonDefaultPositionByBatch(batch, fsplit_set, tree);
// start enumeration
const bst_omp_uint nsize = static_cast<bst_omp_uint>(batch.size);
#pragma omp parallel for schedule(dynamic, 1)