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

Fix inference with categorical feature. (#8591)

Browse Source
This commit is contained in:
Jiaming Yuan
2022-12-15 17:57:26 +08:00
committed by GitHub
parent 7dc3e95a77
commit 43a647a4dd
6 changed files with 75 additions and 28 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
src/tree/updater_gpu_hist.cu
View File

@@ -402,8 +402,7 @@ struct GPUHistMakerDevice {
go_left = data.split_node.DefaultLeft();
} else {
if (data.split_type == FeatureType::kCategorical) {
go_left = common::Decision<false>(data.node_cats.Bits(), cut_value,
data.split_node.DefaultLeft());
go_left = common::Decision(data.node_cats.Bits(), cut_value);
} else {
go_left = cut_value <= data.split_node.SplitCond();
}
@@ -480,7 +479,7 @@ struct GPUHistMakerDevice {
if (common::IsCat(d_feature_types, position)) {
auto node_cats = categories.subspan(categories_segments[position].beg,
categories_segments[position].size);
go_left = common::Decision<false>(node_cats, element, node.DefaultLeft());
go_left = common::Decision(node_cats, element);
} else {
go_left = element <= node.SplitCond();
}