Categorical data support in CPU sketching. (#7221)

2021-09-17 04:37:09 +08:00
parent 9f63d6fead
commit 31c1e13f90
7 changed files with 129 additions and 57 deletions
--- a/src/common/hist_util.h
+++ b/src/common/hist_util.h
@@ -1,5 +1,5 @@
 /*!
- * Copyright 2017-2020 by Contributors
+ * Copyright 2017-2021 by Contributors
 * \file hist_util.h
 * \brief Utility for fast histogram aggregation
 * \author Philip Cho, Tianqi Chen
@@ -128,6 +128,7 @@ inline HistogramCuts SketchOnDMatrix(DMatrix *m, int32_t max_bins,
    }
  }
  HostSketchContainer container(reduced, max_bins,
+                                m->Info().feature_types.ConstHostSpan(),
                                HostSketchContainer::UseGroup(info), threads);
  for (auto const &page : m->GetBatches<SparsePage>()) {
    container.PushRowPage(page, info, hessian);