fix sumup too small when set unbalance=true

0241d323 · Guolin Ke · 5172b533 · 0241d323
Commit 0241d323 authored Nov 05, 2016 by Guolin Ke
Show whitespace changes
Inline Side-by-side

Showing with 7 additions and 2 deletions

src/objective/binary_objective.hpp src/objective/binary_objective.hpp +7 -2

No files found.
--- a/src/objective/binary_objective.hpp
+++ b/src/objective/binary_objective.hpp
@@ -47,8 +47,13 @@ public:
    label_weights_[1] = 1.0f;
    // if using unbalance, change the labels weight
    if (is_unbalance_) {
+      if (cnt_positive > cnt_negative) {
        label_weights_[1] = 1.0f;
        label_weights_[0] = static_cast<score_t>(cnt_positive) / cnt_negative;
+      } else {
+        label_weights_[1] = static_cast<score_t>(cnt_negative) / cnt_positive;
+        label_weights_[0] = 1.0f;
+      }
    }
  }