Vermeer commited on
Commit
bc0d1ff
·
verified ·
1 Parent(s): 5b96629

Upload utils.py

Browse files
Files changed (1) hide show
  1. utils.py +1 -1
utils.py CHANGED
@@ -57,4 +57,4 @@ def predict(model, device, data, num_users, num_items, user_id, train_edge_label
57
  recommended_item = meta_dataframe[meta_dataframe['asin'] == asin_of_item]['title'].values
58
  recommended_items.append(recommended_item)
59
 
60
- return ground_truth_items.to_list(), recommended_items
 
57
  recommended_item = meta_dataframe[meta_dataframe['asin'] == asin_of_item]['title'].values
58
  recommended_items.append(recommended_item)
59
 
60
+ return ground_truth_items.values.tolist(), recommended_items