From d32e7803201c0ac30b9141a311cb783bce9878f1 Mon Sep 17 00:00:00 2001
From: yimutianyang <40798740+yimutianyang@users.noreply.github.com>
Date: Mon, 7 Aug 2023 13:10:26 +0800
Subject: [PATCH] Update rec_dataset.py

---
 rec_dataset.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/rec_dataset.py b/rec_dataset.py
index a43a8e7..8e83962 100644
--- a/rec_dataset.py
+++ b/rec_dataset.py
@@ -34,7 +34,7 @@ def __init__(self, args):
         self.traindata = np.load(self.data_path + 'traindata.npy', allow_pickle=True).tolist()
         self.testdata = np.load(self.data_path + 'testdata.npy', allow_pickle=True).tolist()
         self.num_user, self.num_item = self.max_user_and_item()
-        pdb.set_trace()
+        #pdb.set_trace()
         self.num_node = self.num_user + self.num_item
         self.training_user, self.training_item = [], []
         for u, items in self.traindata.items():
@@ -203,4 +203,4 @@ def _batch_sampling(self, num_negative):
             if index_end == len(indexs):
                 index_start = len(indexs) - self.batch_size
             batch_data = triplet_data[indexs[index_start:index_end]]
-            yield batch_data[:, 0], batch_data[:, 1], batch_data[:, 2]
\ No newline at end of file
+            yield batch_data[:, 0], batch_data[:, 1], batch_data[:, 2]