fix unbalance for binary classification

b3e99b27 · Guolin Ke · 47313fb5 · b3e99b27 · b3e99b27
Commit b3e99b27 authored Nov 01, 2016 by Guolin Ke
Showing with 1140 additions and 2 deletions

examples/binary_classification/LightGBM_model.txt examples/binary_classification/LightGBM_model.txt +1138 -0

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

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