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

Unify the cat split storage for CPU. (#7937)

Browse Source 
* Unify the cat split storage for CPU.

* Cleanup.

* Workaround.
This commit is contained in:
Jiaming Yuan
2022-05-26 19:14:40 +08:00
committed by GitHub
parent 755d9d4609
commit 18cbebaeb9
5 changed files with 25 additions and 33 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
src/tree/tree_model.cc
View File

@@ -808,11 +808,9 @@ void RegTree::ExpandNode(bst_node_t nid, unsigned split_index, bst_float split_v
}
void RegTree::ExpandCategorical(bst_node_t nid, unsigned split_index,
common::Span<uint32_t> split_cat, bool default_left,
bst_float base_weight,
bst_float left_leaf_weight,
bst_float right_leaf_weight,
bst_float loss_change, float sum_hess,
common::Span<const uint32_t> split_cat, bool default_left,
bst_float base_weight, bst_float left_leaf_weight,
bst_float right_leaf_weight, bst_float loss_change, float sum_hess,
float left_sum, float right_sum) {
this->ExpandNode(nid, split_index, std::numeric_limits<float>::quiet_NaN(),
default_left, base_weight,