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

fix

Browse Source
This commit is contained in:
tqchen 2014-09-07 20:01:03 -07:00
parent 19a1ee24a5
commit d4ab359be1
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -282,7 +282,7 @@ class GBTree : public IGradBooster {
tree::RegTree::FVec &feats = thread_temp[tid]; tree::RegTree::FVec &feats = thread_temp[tid];
feats.Fill(batch[i]); feats.Fill(batch[i]);
for (size_t j = 0; j < trees.size(); ++j) { for (size_t j = 0; j < trees.size(); ++j) {
int tid = trees[i]->GetLeafIndex(feats, info.GetRoot(ridx)); int tid = trees[j]->GetLeafIndex(feats, info.GetRoot(ridx));
preds[ridx * mparam.num_trees + j] = static_cast<float>(tid); preds[ridx * mparam.num_trees + j] = static_cast<float>(tid);
} }
feats.Drop(batch[i]); feats.Drop(batch[i]);