From e8665944ca886435faad8c65287711d22e84faf6 Mon Sep 17 00:00:00 2001 From: rishi2019194 <58341663+rishi2019194@users.noreply.github.com> Date: Mon, 16 Oct 2023 14:05:42 -0400 Subject: [PATCH] Added genre tags, imdb_link for predicted movies --- src/prediction_scripts/item_based.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/src/prediction_scripts/item_based.py b/src/prediction_scripts/item_based.py index 8eac546fc..9ed3bf33e 100644 --- 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])