Support categorical data in ellpack. (#6140)

2020-09-24 19:28:57 +08:00
parent 78d72ef936
commit 14afdb4d92
9 changed files with 189 additions and 43 deletions
--- a/tests/cpp/histogram_helpers.h
+++ b/tests/cpp/histogram_helpers.h
@@ -45,7 +45,7 @@ inline std::unique_ptr<EllpackPageImpl> BuildEllpackPage(
  }

  auto page = std::unique_ptr<EllpackPageImpl>(
-      new EllpackPageImpl(0, cmat, batch, dmat->IsDense(), row_stride));
+      new EllpackPageImpl(0, cmat, batch, dmat->IsDense(), row_stride, {}));

  return page;
 }