Support column split in approx tree method (#8847)

2023-03-01 11:59:07 -08:00
parent 6d8afb2218
commit 7cbaee9916
6 changed files with 101 additions and 19 deletions
--- a/src/data/data.cc
+++ b/src/data/data.cc
@@ -912,6 +912,7 @@ DMatrix* DMatrix::Load(const std::string& uri, bool silent, DataSplitMode data_s
    if (!cache_file.empty()) {
      LOG(FATAL) << "Column-wise data split is not support for external memory.";
    }
+    LOG(CONSOLE) << "Splitting data by column";
    auto* sliced = dmat->SliceCol(npart, partid);
    delete dmat;
    return sliced;