Update rec_dataset.py

yimutianyang · Aug 7, 2023 · d32e780 · d32e780
1 parent 73c0c97
commit d32e780
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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]
+            yield batch_data[:, 0], batch_data[:, 1], batch_data[:, 2]