sync for init score of binary objective function (#4332)

Co-authored-by: 未闲 <[email protected]>
microsoft · Jun 25, 2021 · 0701a32 · 0701a32
1 parent 113b44d
commit 0701a32
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/objective/binary_objective.hpp b/src/objective/binary_objective.hpp
@@ -152,6 +152,10 @@ class BinaryLogloss: public ObjectiveFunction {
         suml += is_pos_(label_[i]);
       }
     }
+    if (Network::num_machines() > 1) {
+      suml = Network::GlobalSyncUpBySum(suml);
+      sumw = Network::GlobalSyncUpBySum(sumw);
+    }
     double pavg = suml / sumw;
     pavg = std::min(pavg, 1.0 - kEpsilon);
     pavg = std::max<double>(pavg, kEpsilon);