From 84d3fcb7ea8f4a02a3217e74c2b16660a6e387ae Mon Sep 17 00:00:00 2001 From: ZHAOKAI WANG Date: Thu, 8 Jun 2023 01:24:04 +0800 Subject: [PATCH] Fix `cpu_predictor` categorical feature disaptch (#9256) --- src/predictor/cpu_predictor.cc | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/predictor/cpu_predictor.cc b/src/predictor/cpu_predictor.cc index 96c1fbe18..920e81dcd 100644 --- a/src/predictor/cpu_predictor.cc +++ b/src/predictor/cpu_predictor.cc @@ -152,7 +152,7 @@ void PredictByAllTrees(gbm::GBTreeModel const &model, std::uint32_t const tree_b } else { for (std::size_t i = 0; i < block_size; ++i) { out_predt(predict_offset + i, gid) += - scalar::PredValueByOneTree(thread_temp[offset + i], tree, cats); + scalar::PredValueByOneTree(thread_temp[offset + i], tree, cats); } } }