diff --git a/docs/GettingStarted.ipynb b/docs/GettingStarted.ipynb index 7053e162b..9742b32e7 100644 --- 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",