update getting-started for dataset

lenskit · Jul 23, 2024 · e9bb3ea · e9bb3ea
1 parent a18dc77
commit e9bb3ea
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/docs/GettingStarted.ipynb b/docs/GettingStarted.ipynb
@@ -27,6 +27,7 @@
    "outputs": [],
    "source": [
     "from lenskit.datasets import ML100K\n",
+    "from lenskit.data import from_interactions_df\n",
     "from lenskit import batch, topn, util\n",
     "from lenskit import crossfold as xf\n",
     "from lenskit.algorithms import Recommender, als, knn\n",
@@ -216,7 +217,7 @@
     "def eval(aname, algo, train, test):\n",
     "    fittable = util.clone(algo)\n",
     "    fittable = Recommender.adapt(fittable)\n",
-    "    fittable.fit(train)\n",
+    "    fittable.fit(from_interactions_df(train))\n",
     "    users = test.user.unique()\n",
     "    # now we run the recommender\n",
     "    recs = batch.recommend(fittable, users, 100)\n",