Support categorical split in tree model dump. (#7036)

2021-06-18 16:46:20 +08:00
parent 7968c0d051
commit 29f8fd6fee
8 changed files with 263 additions and 46 deletions
--- a/src/predictor/cpu_predictor.cc
+++ b/src/predictor/cpu_predictor.cc
@@ -52,11 +52,6 @@ bst_float PredValue(const SparsePage::Inst &inst,
    if (tree_info[i] == bst_group) {
      auto const &tree = *trees[i];
      bool has_categorical = tree.HasCategoricalSplit();
-
-      auto categories = common::Span<uint32_t const>{tree.GetSplitCategories()};
-      auto split_types = tree.GetSplitTypes();
-      auto categories_ptr =
-          common::Span<RegTree::Segment const>{tree.GetSplitCategoriesPtr()};
      auto cats = tree.GetCategoriesMatrix();
      bst_node_t nidx = -1;
      if (has_categorical) {