diff --git a/recbole/model/general_recommender/line.py b/recbole/model/general_recommender/line.py index 7c183c2f4..9a7fbf9f3 100644 --- a/recbole/model/general_recommender/line.py +++ b/recbole/model/general_recommender/line.py @@ -69,7 +69,7 @@ def __init__(self, config, dataset): def get_used_ids(self): cur = np.array([set() for _ in range(self.n_items)]) - for iid, uid in zip(self.interaction_feat[self.USER_ID].numpy(), self.interaction_feat[self.ITEM_ID].numpy()): + for uid, iid in zip(self.interaction_feat[self.USER_ID].numpy(), self.interaction_feat[self.ITEM_ID].numpy()): cur[iid].add(uid) return cur