More refactoring to take advantage of collective aggregators (#9081)

2023-04-25 12:36:09 -07:00
parent 49ccae7fb9
commit a320b402a5
10 changed files with 81 additions and 81 deletions
--- a/include/xgboost/data.h
+++ b/include/xgboost/data.h
@@ -196,6 +196,14 @@ class MetaInfo {
   */
  bool IsVerticalFederated() const;

+  /*!
+   * \brief A convenient method to check if the MetaInfo should contain labels.
+   *
+   * Normally we assume labels are available everywhere. The only exception is in vertical federated
+   * learning where labels are only available on worker 0.
+   */
+  bool ShouldHaveLabels() const;
+
 private:
  void SetInfoFromHost(Context const& ctx, StringView key, Json arr);
  void SetInfoFromCUDA(Context const& ctx, StringView key, Json arr);