Add max_cat_threshold to GPU and handle missing cat values. (#8212)

2022-09-07 00:57:51 +08:00
parent 441ffc017a
commit b5eb36f1af
10 changed files with 546 additions and 122 deletions
--- a/src/tree/gpu_hist/evaluate_splits.cuh
+++ b/src/tree/gpu_hist/evaluate_splits.cuh
@@ -141,7 +141,8 @@ class GPUHistEvaluator {
   */
  common::Span<CatST const> GetHostNodeCats(bst_node_t nidx) const {
    copy_stream_.View().Sync();
-    auto cats_out = common::Span<CatST const>{h_split_cats_}.subspan(nidx * node_categorical_storage_size_, node_categorical_storage_size_);
+    auto cats_out = common::Span<CatST const>{h_split_cats_}.subspan(
+        nidx * node_categorical_storage_size_, node_categorical_storage_size_);
    return cats_out;
  }
  /**