fix error

morningsky · May 23, 2022 · afa33f1 · afa33f1
1 parent 7c30fad
commit afa33f1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/examples/matching/run_movielens.py b/examples/matching/run_movielens.py
@@ -138,7 +138,7 @@ def main(dataset_path, model_name, epoch, learning_rate, batch_size, weight_deca
     user_col = "user_id"
     item_col = "movie_id"
 
-    data = pd.DataFrame({"user_id": all_item["user_id"], "movie_id": all_item["movie_id"]})
+    data = pd.DataFrame({"user_id": test_user["user_id"], "movie_id": test_user["movie_id"]})
     data[user_col] = data[user_col].map(user_map)
     data[item_col] = data[item_col].map(item_map)
     user_pos_item = data.groupby(user_col).agg(list).reset_index()