Skip to content

Commit

Permalink
Merge pull request #1224 from zhaisw/Issue1222_SurpriseUtils
Browse files Browse the repository at this point in the history
Updated surprise_utils.py
  • Loading branch information
miguelgfierro authored Oct 29, 2020
2 parents 3402979 + d392469 commit 675be76
Showing 1 changed file with 5 additions and 2 deletions.
7 changes: 5 additions & 2 deletions reco_utils/recommender/surprise/surprise_utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -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])
Expand Down

0 comments on commit 675be76

Please sign in to comment.