Merge branch 'master' into dev

f3b1daf8 · Guolin Ke · dd5f2b27 · 0241d323 · f3b1daf8
Commit f3b1daf8 authored Nov 05, 2016 by Guolin Ke
Hide 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_) {
-      label_weights_[1] = 1.0f;
+      if (cnt_positive > cnt_negative) {
-      label_weights_[0] = static_cast<score_t>(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;
+      }
    }
  }