Fix undefined name in AUCMetric

apache · Aug 13, 2018 · 6505674 · 6505674
1 parent bf81dcb
commit 6505674
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/example/sparse/factorization_machine/metric.py b/example/sparse/factorization_machine/metric.py
@@ -107,7 +107,9 @@ def update(self, labels, preds):
         label_sum = label_weight.sum()
         if label_sum == 0 or label_sum == label_weight.size:
             raise Exception("AUC with one class is undefined")
-
+
+        label_one_num = np.count_nonzero(label_weight)
+        label_zero_num = len(label_weight) - label_one_num
         total_area = label_zero_num * label_one_num
         height = 0
         width = 0