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

add changes

Browse Source
This commit is contained in:
tqchen
2014-08-24 17:25:17 -07:00
parent da75f8f1a4
commit 7874c2559b
5 changed files with 52 additions and 28 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/gbm/gbtree-inl.hpp
View File

@@ -135,7 +135,7 @@ class GBTree : public IGradBooster<FMatrix> {
const int tid = omp_get_thread_num();
tree::RegTree::FVec &feats = thread_temp[tid];
const size_t ridx = batch.base_rowid + i;
const unsigned root_idx = info.GetRoot(i);
const unsigned root_idx = info.GetRoot(ridx);
// loop over output groups
for (int gid = 0; gid < mparam.num_output_group; ++gid) {
preds[ridx * mparam.num_output_group + gid] =