Update metrics.py

muxujie · Aug 7, 2021 · 609b153 · 609b153
1 parent 22069e0
commit 609b153
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/metrics.py b/metrics.py
@@ -98,7 +98,7 @@ def auc(ground_truth, prediction):
 
 def mrr(r):
     r = np.array(r)
-    if np.sum(r) == 0.
+    if np.sum(r) == 0.:
         return 0.
     else:
         return np.reciprocal(np.where(r==1)[0]+1, dtype=np.float)[0]