fix bug in multi_error

8aa15e88 · Guolin Ke · db18367e · 8aa15e88
Commit 8aa15e88 authored Dec 23, 2016 by Guolin Ke
Show whitespace changes
Inline Side-by-side

Showing with 3 additions and 3 deletions

src/metric/multiclass_metric.hpp src/metric/multiclass_metric.hpp +3 -3

No files found.
--- a/src/metric/multiclass_metric.hpp
+++ b/src/metric/multiclass_metric.hpp
@@ -101,11 +101,11 @@ public:
  inline static score_t LossOnPoint(float label, std::vector<double> score) {
    size_t k = static_cast<size_t>(label);
    for (size_t i = 0; i < score.size(); ++i){
-        if (i != k && score[i] > score[k]) {
+        if (i != k && score[i] >= score[k]) {
-            return 0.0f;
+            return 1.0f;
        }
    }
-    return 1.0f;
+    return 0.0f;
  }
  inline static const char* Name() {