From 2a662c848fb24fb561240f50fb6a2727ec61a4ef Mon Sep 17 00:00:00 2001
From: Ivan Maksimov <ivan.maksimov@corp.mail.ru>
Date: Thu, 4 Jun 2020 21:57:55 +0300
Subject: [PATCH] added recall@k

---
 src/metrics.py | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

diff --git a/src/metrics.py b/src/metrics.py
index 9ebcc0d..47984c0 100644
--- 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