shorten argument name

PreferredAI · Dec 9, 2023 · 94c1fe2 · 94c1fe2
1 parent 8634db3
commit 94c1fe2
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/cornac/eval_methods/next_basket_evaluation.py b/cornac/eval_methods/next_basket_evaluation.py
@@ -128,7 +128,7 @@ def get_gt_items(train_set, test_set, test_pos_items, exclude_unknowns):
             user_idx,
             item_indices,
             history_baskets=history_baskets,
-            history_basket_indices=bids[:-1],
+            history_bids=bids[:-1],
             uir_tuple=test_set.uir_tuple,
             baskets=test_set.baskets,
             basket_indices=test_set.basket_indices,

diff --git a/cornac/models/gp_top/recom_gp_top.py b/cornac/models/gp_top/recom_gp_top.py
@@ -82,11 +82,11 @@ def score(self, user_idx, history_baskets, **kwargs):
 
         if self.use_personalized_popularity:
             if self.use_quantity:
-                history_basket_indices = kwargs.get("history_basket_indices")
+                history_bids = kwargs.get("history_bids")
                 baskets = kwargs.get("baskets")
                 p_item_freq = Counter()
                 extra_data = kwargs.get("extra_data")
-                for bid, iids in zip(history_basket_indices, history_baskets):
+                for bid, iids in zip(history_bids, history_baskets):
                     for idx, iid in zip(baskets[bid], iids):
                         p_item_freq[iid] += extra_data[idx].get("quantity", 0)
             else: