Initial support for column-wise data split (#8468)

2022-12-03 09:37:51 -08:00
parent c0609b98f1
commit 78d65a1928
8 changed files with 135 additions and 3 deletions
--- a/src/data/proxy_dmatrix.h
+++ b/src/data/proxy_dmatrix.h
@@ -87,6 +87,10 @@ class DMatrixProxy : public DMatrix {
    LOG(FATAL) << "Slicing DMatrix is not supported for Proxy DMatrix.";
    return nullptr;
  }
+  DMatrix* SliceCol(std::size_t start, std::size_t size) override {
+    LOG(FATAL) << "Slicing DMatrix columns is not supported for Proxy DMatrix.";
+    return nullptr;
+  }
  BatchSet<SparsePage> GetRowBatches() override {
    LOG(FATAL) << "Not implemented.";
    return BatchSet<SparsePage>(BatchIterator<SparsePage>(nullptr));