Fix SAR notebooks

recommenders-team · Jun 9, 2022 · 68b60c0 · 68b60c0
1 parent d3d4b98
commit 68b60c0
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 6 deletions.
diff --git a/examples/00_quick_start/sar_movielens.ipynb b/examples/00_quick_start/sar_movielens.ipynb
@@ -402,7 +402,7 @@
    ],
    "source": [
     "with Timer() as test_time:\n",
-    "    top_k = model.recommend_k_items(test, remove_seen=True)\n",
+    "    top_k = model.recommend_k_items(test, top_k=TOP_K, remove_seen=True)\n",
     "\n",
     "print(\"Took {} seconds for prediction.\".format(test_time.interval))"
    ]
@@ -923,4 +923,4 @@
  },
  "nbformat": 4,
  "nbformat_minor": 4
-}
+}
diff --git a/examples/00_quick_start/sar_movielens_with_azureml.ipynb b/examples/00_quick_start/sar_movielens_with_azureml.ipynb
@@ -400,7 +400,7 @@
     "\n",
     "# predict top k items\n",
     "with Timer() as t:\n",
-    "    top_k = model.recommend_k_items(test, remove_seen=True)\n",
+    "    top_k = model.recommend_k_items(test, top_k=TOP_K, remove_seen=True)\n",
     "\n",
     "run.log(name=\"Prediction time\", value=t.interval)\n",
     "\n",
@@ -693,4 +693,4 @@
  },
  "nbformat": 4,
  "nbformat_minor": 2
-}
+}
diff --git a/examples/02_model_collaborative_filtering/sar_deep_dive.ipynb b/examples/02_model_collaborative_filtering/sar_deep_dive.ipynb
@@ -121,7 +121,6 @@
    "source": [
     "# set the environment path to find Recommenders\n",
     "import sys\n",
-
     "\n",
     "import itertools\n",
     "import logging\n",
@@ -390,7 +389,7 @@
     }
    ],
    "source": [
-    "top_k = model.recommend_k_items(test, remove_seen=True)"
+    "top_k = model.recommend_k_items(test, top_k=TOP_K, remove_seen=True)"
    ]
   },
   {