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

Co-authored-by: 未闲 <weixian.lzf@antfin.com>

sync for init score of binary objective function (#4332)
Co-authored-by: 未闲 <weixian.lzf@antfin.com>
0701a32d · Arcs · GitHub · 113b44d7 · 0701a32d
Unverified Commit 0701a32d authored Jun 25, 2021 by Arcs Committed by GitHub Jun 24, 2021
Show whitespace changes
Inline Side-by-side

Showing with 4 additions and 0 deletions

src/objective/binary_objective.hpp src/objective/binary_objective.hpp +4 -0

No files found.
--- 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);