added recall@k

geangohn · Jun 4, 2020 · 2a662c8 · 2a662c8
1 parent df89619
commit 2a662c8
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/src/metrics.py b/src/metrics.py
@@ -48,7 +48,15 @@ def recall(recommended_list, bought_list):
 
 
 def recall_at_k(recommended_list, bought_list, k=5):
-    # your_code
+
+    bought_list = np.array(bought_list)
+    recommended_list = np.array(recommended_list)
+
+    if k < len(recommended_list):
+        recommended_list = recommended_list[:k]
+
+    flags = np.isin(bought_list, recommended_list)
+    recall = flags.sum() / len(bought_list)
 
     return recall