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

Update engine_robust.cc

Browse Source
This commit is contained in:
Tianqi Chen 2014-12-01 15:32:57 -08:00
parent 1c5167d96e
commit fd2c57b8a4
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
src/engine_robust.cc
View File

@ -303,7 +303,8 @@ ShortestDist(const std::pair<bool, size_t> &node_value,
res = dist_in[i].first; size = dist_in[i].second; res = dist_in[i].first; size = dist_in[i].second;
} }
} }
return std::make_pair(res, size); // add one hop
return std::make_pair(res + 1, size);
} }
/*! /*!
* \brief message passing function, used to decide the * \brief message passing function, used to decide the