Added genre tags, imdb_link for predicted movies

adipai · Oct 16, 2023 · e866594 · e866594
1 parent d447188
commit e866594
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/prediction_scripts/item_based.py b/src/prediction_scripts/item_based.py
@@ -49,4 +49,6 @@ def recommend_for_new_user(user_rating):
         by="recommended", ascending=False, inplace=True
     )
 
-    return list(join_movies_and_recommendations["title"][:201])
+    return list(join_movies_and_recommendations["title"][:201]), \
+        list(join_movies_and_recommendations["genres"][:201]), \
+            list(join_movies_and_recommendations["imbd_id"][:201])