FIX use np.bincount

arjoly · arjoly · commit 4d9b730b9de9 · 2014-04-03T08:04:23.000+02:00
diff --git a/sklearn/metrics/metrics.py b/sklearn/metrics/metrics.py
@@ -2079,14 +2079,14 @@ def label_ranking_average_precision_score(y_true, y_score):
 
         # Compute denominators
         unique_all, inverse = np.unique(y_score[i], return_inverse=True)
-        count = bincount(inverse, minlength=unique_all.size)
+        count = np.bincount(inverse, minlength=unique_all.size)
         cum_count = count[::-1].cumsum()[::-1]  # reverse cumsum
 
         # Compute numerators
         unique_relevant, relevant_inverse = np.unique(y_score[i, relevant],
                                                       return_inverse=True)
-        count_relevant = bincount(relevant_inverse,
-                                  minlength=unique_relevant.size)
+        count_relevant = np.bincount(relevant_inverse,
+                                     minlength=unique_relevant.size)
         cum_count_relevant = count_relevant[::-1].cumsum()[::-1]
 
         score += (cum_count_relevant[relevant_inverse] /
