Spaces:
Sleeping
Sleeping
Upload utils.py
Browse files
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.
|
|
|
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
|