Update AMF.py

hexiangnan · Oct 11, 2018 · 61a4ee2 · 61a4ee2
1 parent fee12e4
commit 61a4ee2
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/AMF.py b/AMF.py
@@ -115,7 +115,7 @@ def _get_train_batch(i):
             # negtive k
             gtItem = _dataset.testRatings[user][1]
             j = np.random.randint(_dataset.num_items)
-            while j in _dataset.trainList[_user_input[_index[idx]]] or j == gtItem:
+            while j in _dataset.trainList[_user_input[_index[idx]]]:
                 j = np.random.randint(_dataset.num_items)
             item_neg_batch.append(j)
     return np.array(user_batch)[:, None], np.array(item_batch)[:, None], \
@@ -529,4 +529,4 @@ def init_logging(args, time_stamp):
     print "Initialize AMF"
 
     # start training
-    training(AMF, dataset, args, epoch_start=args.adv_epoch, epoch_end=args.epochs, time_stamp=time_stamp)
+    training(AMF, dataset, args, epoch_start=args.adv_epoch, epoch_end=args.epochs, time_stamp=time_stamp)