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

Extract CPUExpandEntry and HistParam. (#7321)

Browse Source 
* Remove kRootNid.
* Check for empty hessian.
This commit is contained in:
Jiaming Yuan
2021-10-17 14:22:25 +08:00
committed by GitHub
parent 6cdcfe8128
commit 8e619010d0
5 changed files with 93 additions and 48 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/cpp/tree/hist/test_histogram.cc
View File

@@ -258,7 +258,7 @@ void TestBuildHistogram(bool is_distributed) {
std::iota(row_indices.begin(), row_indices.end(), 0);
row_set_collection_.Init();
CPUExpandEntry node(CPUExpandEntry::kRootNid, tree.GetDepth(0), 0.0f);
CPUExpandEntry node(RegTree::kRoot, tree.GetDepth(0), 0.0f);
std::vector<CPUExpandEntry> nodes_for_explicit_hist_build_;
nodes_for_explicit_hist_build_.push_back(node);
histogram.BuildHist(p_fmat.get(), &tree, row_set_collection_,