label is integer

apache · Feb 28, 2017 · 681e7fb · 681e7fb
1 parent 1d9d71b
commit 681e7fb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/core/src/main/java/hivemall/evaluation/AUCUDAF.java b/core/src/main/java/hivemall/evaluation/AUCUDAF.java
@@ -297,7 +297,7 @@ void iterate(double score, int label) {
                 fpPrev = fp;
                 tpPrev = tp;
             }
-            if (label == 1.d) {
+            if (label == 1) {
                 tp += 1; // this finally will be the number of positive samples
             } else {
                 fp += 1; // this finally will be the number of negative samples