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/include/xgboost/data.h
+++ b/include/xgboost/data.h
@@ -529,6 +529,11 @@ class DMatrix {
    return Info().num_nonzero_ == Info().num_row_ * Info().num_col_;
  }

+  /*! \brief Whether the data is split row-wise. */
+  bool IsRowSplit() const {
+    return Info().data_split_mode == DataSplitMode::kRow;
+  }
+
  /*! \brief Whether the data is split column-wise. */
  bool IsColumnSplit() const {
    return Info().data_split_mode == DataSplitMode::kCol;