Implement weighted sketching for adapter. (#5760)

* Bounded memory tests. * Fixed memory estimation.
2020-06-12 06:20:39 +08:00
parent c35be9dc40
commit 3028fa6b42
7 changed files with 443 additions and 109 deletions
--- a/src/common/hist_util.cc
+++ b/src/common/hist_util.cc
@@ -140,6 +140,10 @@ void HistogramCuts::Build(DMatrix* dmat, uint32_t const max_num_bins) {

 bool CutsBuilder::UseGroup(DMatrix* dmat) {
  auto& info = dmat->Info();
+  return CutsBuilder::UseGroup(info);
+}
+
+bool CutsBuilder::UseGroup(MetaInfo const& info) {
  size_t const num_groups = info.group_ptr_.size() == 0 ?
                            0 : info.group_ptr_.size() - 1;
  // Use group index for weights?