From d392469d362708807125f82da5977d7b15bf28d5 Mon Sep 17 00:00:00 2001 From: zhaisw Date: Thu, 29 Oct 2020 21:55:04 +1300 Subject: [PATCH] Updated surprise_utils.py --- reco_utils/recommender/surprise/surprise_utils.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/reco_utils/recommender/surprise/surprise_utils.py b/reco_utils/recommender/surprise/surprise_utils.py index ad9f702b1e..21c769910f 100644 --- a/reco_utils/recommender/surprise/surprise_utils.py +++ b/reco_utils/recommender/surprise/surprise_utils.py @@ -96,8 +96,11 @@ def compute_ranking_predictions( pd.DataFrame: dataframe with usercol, itemcol, predcol """ preds_lst = [] - for user in data[usercol].unique(): - for item in data[itemcol].unique(): + users = data[usercol].unique() + items = data[itemcol].unique() + + for user in users: + for item in items: preds_lst.append([user, item, algo.predict(user, item).est]) all_predictions = pd.DataFrame(data=preds_lst, columns=[usercol, itemcol, predcol])