Merge pull request #1651 from ywkim92/patch-1

fix: Change formats of user_ids and item_ids arg.
recommenders-team · Feb 19, 2022 · 6c53f92 · 6c53f92
2 parents 4639087 + 6149723
commit 6c53f92
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/recommenders/models/lightfm/lightfm_utils.py b/recommenders/models/lightfm/lightfm_utils.py
@@ -253,8 +253,8 @@ def prepare_all_predictions(
 
     all_predictions["prediction"] = all_predictions.apply(
         lambda x: model.predict(
-            user_ids=int(x["uid"]),
-            item_ids=[x["iid"]],
+            user_ids=np.array([x["uid"]], dtype=np.int32),
+            item_ids=np.array([x["iid"]], dtype=np.int32),
             user_features=user_features,
             item_features=item_features,
             num_threads=num_threads,