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

Optimized EvaluateSplut function (#5138)

Browse Source 
* Add block based threading utilities.
This commit is contained in:
Egor Smirnov
2019-12-31 13:18:42 +03:00
committed by Jiaming Yuan
parent 04db125699
commit 7b17e76c5b
5 changed files with 317 additions and 47 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
tests/cpp/tree/test_quantile_hist.cc
View File

@@ -211,7 +211,8 @@ class QuantileHistMock : public QuantileHistMaker {
}
/* Now compare against result given by EvaluateSplit() */
RealImpl::EvaluateSplit(0, gmat, hist_, *(*dmat), tree);
ExpandEntry node(0, tree.GetDepth(0), snode_[0].best.loss_chg, 0);
RealImpl::EvaluateSplit({node}, gmat, hist_, *(*dmat), tree);
ASSERT_EQ(snode_[0].best.SplitIndex(), best_split_feature);
ASSERT_EQ(snode_[0].best.split_value, gmat.cut.Values()[best_split_threshold]);