Merge pull request #9 from mmaorc/fix-movie-average-calculation-example

Fix 'Toy Story' movie average calculation bug in Ex8
2018-10-08 22:03:20 -04:00
parent f531bf28d2 07bf4dd931
commit 40fd933e36
1 changed files with 1 additions and 1 deletions
--- a/Exercise8/exercise8.ipynb
+++ b/Exercise8/exercise8.ipynb
@@ -468,7 +468,7 @@
    "\n",
    "# From the matrix, we can compute statistics like average rating.\n",
    "print('Average rating for movie 1 (Toy Story): %f / 5' %\n",
-    "      np.mean(Y[0, R[0, :]]))\n",
+    "      np.mean(Y[0, R[0, :] == 1]))\n",
    "\n",
    "# We can \"visualize\" the ratings matrix by plotting it with imshow\n",
    "pyplot.figure(figsize=(8, 8))\n",