MovieRec-CollabFilt-RBM.ipynb

{
 "cells": [
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "<a id=\"ref1\"></a>\n",
    "# Acquiring the Data"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "To start, we need to download the data we are going to use for our system. The datasets we're going to use were acquired by [GroupLens](http://grouplens.org/datasets/movielens/) and contain movies, users and movie ratings by these users.\n",
    "\n",
    "After the download is done, we extract the datasets to a directory that's easily accessible."
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 1,
   "metadata": {
    "collapsed": false
   },
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "--2017-04-07 01:34:43--  http://files.grouplens.org/datasets/movielens/ml-1m.zip\n",
      "Resolving files.grouplens.org (files.grouplens.org)... 128.101.34.146\n",
      "Connecting to files.grouplens.org (files.grouplens.org)|128.101.34.146|:80... connected.\n",
      "HTTP request sent, awaiting response... 200 OK\n",
      "Length: 5917549 (5.6M) [application/zip]\n",
      "Saving to: ‘moviedataset.zip’\n",
      "\n",
      "100%[======================================>] 5,917,549   21.4MB/s   in 0.3s   \n",
      "\n",
      "2017-04-07 01:34:43 (21.4 MB/s) - ‘moviedataset.zip’ saved [5917549/5917549]\n",
      "\n",
      "Archive:  moviedataset.zip\n",
      "   creating: /resources/data/ml-1m/\n",
      "  inflating: /resources/data/ml-1m/movies.dat  \n",
      "  inflating: /resources/data/ml-1m/ratings.dat  \n",
      "  inflating: /resources/data/ml-1m/README  \n",
      "  inflating: /resources/data/ml-1m/users.dat  \n"
     ]
    }
   ],
   "source": [
    "!wget -O moviedataset.zip http://files.grouplens.org/datasets/movielens/ml-1m.zip\n",
    "!unzip -o moviedataset.zip -d /resources/data"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {
    "collapsed": true
   },
   "source": [
    "With the datasets in place, let's now import the necessary libraries. We will be using [Tensorflow](https://www.tensorflow.org/) and [Numpy](http://www.numpy.org/) together to model and initialize our Restricted Boltzmann Machine and [Pandas](http://pandas.pydata.org/pandas-docs/stable/) to manipulate our datasets. To import these libraries, run the code cell below."
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 2,
   "metadata": {
    "collapsed": false
   },
   "outputs": [],
   "source": [
    "#Tensorflow library. Used to implement machine learning models\n",
    "import tensorflow as tf\n",
    "#Numpy contains helpful functions for efficient mathematical calculations\n",
    "import numpy as np\n",
    "#Dataframe manipulation library\n",
    "import pandas as pd\n",
    "#Graph plotting library\n",
    "import matplotlib.pyplot as plt\n",
    "%matplotlib inline"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "-------------"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "<a id=\"ref2\"></a>\n",
    "# Loading in the Data\n",
    "\n",
    "Let's begin by loading in our data with Pandas. The .dat files containing our data are similar to CSV files, but instead of using the ',' (comma) character to separate entries, it uses '::' (two colons) characters instead. To let Pandas know that it should separate data points at every '::', we have to specify the `sep='::'` parameter when calling the function.\n",
    "\n",
    "Additionally, we also pass it the `header=None` parameter due to the fact that our files don't contain any headers.\n",
    "\n",
    "Let's start with the movies.dat file and take a look at its structure:"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 3,
   "metadata": {
    "collapsed": false,
    "scrolled": true
   },
   "outputs": [
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "/usr/local/lib/python2.7/dist-packages/ipykernel/__main__.py:2: ParserWarning: Falling back to the 'python' engine because the 'c' engine does not support regex separators (separators > 1 char and different from '\\s+' are interpreted as regex); you can avoid this warning by specifying engine='python'.\n",
      "  from ipykernel import kernelapp as app\n"
     ]
    },
    {
     "data": {
      "text/html": [
       "<div>\n",
       "<table border=\"1\" class=\"dataframe\">\n",
       "  <thead>\n",
       "    <tr style=\"text-align: right;\">\n",
       "      <th></th>\n",
       "      <th>0</th>\n",
       "      <th>1</th>\n",
       "      <th>2</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>1</td>\n",
       "      <td>Toy Story (1995)</td>\n",
       "      <td>Animation|Children's|Comedy</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>2</td>\n",
       "      <td>Jumanji (1995)</td>\n",
       "      <td>Adventure|Children's|Fantasy</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>3</td>\n",
       "      <td>Grumpier Old Men (1995)</td>\n",
       "      <td>Comedy|Romance</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>4</td>\n",
       "      <td>Waiting to Exhale (1995)</td>\n",
       "      <td>Comedy|Drama</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>5</td>\n",
       "      <td>Father of the Bride Part II (1995)</td>\n",
       "      <td>Comedy</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "   0                                   1                             2\n",
       "0  1                    Toy Story (1995)   Animation|Children's|Comedy\n",
       "1  2                      Jumanji (1995)  Adventure|Children's|Fantasy\n",
       "2  3             Grumpier Old Men (1995)                Comedy|Romance\n",
       "3  4            Waiting to Exhale (1995)                  Comedy|Drama\n",
       "4  5  Father of the Bride Part II (1995)                        Comedy"
      ]
     },
     "execution_count": 3,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "#Loading in the movies dataset\n",
    "movies_df = pd.read_csv('/resources/data/ml-1m/movies.dat', sep='::', header=None)\n",
    "movies_df.head()"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "We can do the same for the ratings.dat file:"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 4,
   "metadata": {
    "collapsed": false,
    "scrolled": false
   },
   "outputs": [
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "/usr/local/lib/python2.7/dist-packages/ipykernel/__main__.py:2: ParserWarning: Falling back to the 'python' engine because the 'c' engine does not support regex separators (separators > 1 char and different from '\\s+' are interpreted as regex); you can avoid this warning by specifying engine='python'.\n",
      "  from ipykernel import kernelapp as app\n"
     ]
    },
    {
     "data": {
      "text/html": [
       "<div>\n",
       "<table border=\"1\" class=\"dataframe\">\n",
       "  <thead>\n",
       "    <tr style=\"text-align: right;\">\n",
       "      <th></th>\n",
       "      <th>0</th>\n",
       "      <th>1</th>\n",
       "      <th>2</th>\n",
       "      <th>3</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>1</td>\n",
       "      <td>1193</td>\n",
       "      <td>5</td>\n",
       "      <td>978300760</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>1</td>\n",
       "      <td>661</td>\n",
       "      <td>3</td>\n",
       "      <td>978302109</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>1</td>\n",
       "      <td>914</td>\n",
       "      <td>3</td>\n",
       "      <td>978301968</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>1</td>\n",
       "      <td>3408</td>\n",
       "      <td>4</td>\n",
       "      <td>978300275</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>1</td>\n",
       "      <td>2355</td>\n",
       "      <td>5</td>\n",
       "      <td>978824291</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "   0     1  2          3\n",
       "0  1  1193  5  978300760\n",
       "1  1   661  3  978302109\n",
       "2  1   914  3  978301968\n",
       "3  1  3408  4  978300275\n",
       "4  1  2355  5  978824291"
      ]
     },
     "execution_count": 4,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "#Loading in the ratings dataset\n",
    "ratings_df = pd.read_csv('/resources/data/ml-1m/ratings.dat', sep='::', header=None)\n",
    "ratings_df.head()"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "So our movies_df variable contains a dataframe that stores a movie's unique ID number, title and genres, while our ratings_df variable stores a unique User ID number, a movie's ID that the user has watched, the user's rating to said movie and when the user rated that movie.\n",
    "\n",
    "Let's now rename the columns in these dataframes so we can better convey their data more intuitively:"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 5,
   "metadata": {
    "collapsed": false
   },
   "outputs": [],
   "source": [
    "movies_df.columns = ['MovieID', 'Title', 'Genres']\n",
    "ratings_df.columns = ['UserID', 'MovieID', 'Rating', 'Timestamp']"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "Here's our final movies_df:"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 6,
   "metadata": {
    "collapsed": false,
    "scrolled": true
   },
   "outputs": [
    {
     "data": {
      "text/html": [
       "<div>\n",
       "<table border=\"1\" class=\"dataframe\">\n",
       "  <thead>\n",
       "    <tr style=\"text-align: right;\">\n",
       "      <th></th>\n",
       "      <th>MovieID</th>\n",
       "      <th>Title</th>\n",
       "      <th>Genres</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>1</td>\n",
       "      <td>Toy Story (1995)</td>\n",
       "      <td>Animation|Children's|Comedy</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>2</td>\n",
       "      <td>Jumanji (1995)</td>\n",
       "      <td>Adventure|Children's|Fantasy</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>3</td>\n",
       "      <td>Grumpier Old Men (1995)</td>\n",
       "      <td>Comedy|Romance</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>4</td>\n",
       "      <td>Waiting to Exhale (1995)</td>\n",
       "      <td>Comedy|Drama</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>5</td>\n",
       "      <td>Father of the Bride Part II (1995)</td>\n",
       "      <td>Comedy</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "   MovieID                               Title                        Genres\n",
       "0        1                    Toy Story (1995)   Animation|Children's|Comedy\n",
       "1        2                      Jumanji (1995)  Adventure|Children's|Fantasy\n",
       "2        3             Grumpier Old Men (1995)                Comedy|Romance\n",
       "3        4            Waiting to Exhale (1995)                  Comedy|Drama\n",
       "4        5  Father of the Bride Part II (1995)                        Comedy"
      ]
     },
     "execution_count": 6,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "movies_df.head()"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "And our final ratings_df:"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 7,
   "metadata": {
    "collapsed": false,
    "scrolled": true
   },
   "outputs": [
    {
     "data": {
      "text/html": [
       "<div>\n",
       "<table border=\"1\" class=\"dataframe\">\n",
       "  <thead>\n",
       "    <tr style=\"text-align: right;\">\n",
       "      <th></th>\n",
       "      <th>UserID</th>\n",
       "      <th>MovieID</th>\n",
       "      <th>Rating</th>\n",
       "      <th>Timestamp</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>1</td>\n",
       "      <td>1193</td>\n",
       "      <td>5</td>\n",
       "      <td>978300760</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>1</td>\n",
       "      <td>661</td>\n",
       "      <td>3</td>\n",
       "      <td>978302109</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>1</td>\n",
       "      <td>914</td>\n",
       "      <td>3</td>\n",
       "      <td>978301968</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>1</td>\n",
       "      <td>3408</td>\n",
       "      <td>4</td>\n",
       "      <td>978300275</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>1</td>\n",
       "      <td>2355</td>\n",
       "      <td>5</td>\n",
       "      <td>978824291</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "   UserID  MovieID  Rating  Timestamp\n",
       "0       1     1193       5  978300760\n",
       "1       1      661       3  978302109\n",
       "2       1      914       3  978301968\n",
       "3       1     3408       4  978300275\n",
       "4       1     2355       5  978824291"
      ]
     },
     "execution_count": 7,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "ratings_df.head()"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "-----------"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "<a id=\"ref3\"></a>\n",
    "# The Restricted Boltzmann Machine model"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "<img src=\"https://ibm.box.com/shared/static/o049tx0dsllpbj3b546vuba25qqlzelq.png\" alt=\"RBM Model\" style=\"width: 300px;\"/>\n",
    "\n",
    "The Restricted Boltzmann Machine model has two layers of neurons, one of which is what we call a visible input layer and the other is called a hidden layer. The hidden layer is used to learn features from the information fed through the input layer. For our model, the input is going to contain X neurons, where X is the amount of movies in our dataset. Each of these neurons will possess a normalized rating value varying from 0 to 1 -- 0 meaning that a user has not watched that movie and the closer the value is to 1, the more the user likes the movie that neuron's representing. These normalized values, of course, will be extracted and normalized from the ratings dataset.\n",
    "\n",
    "After passing in the input, we train the RBM on it and have the hidden layer learn its features. These features are what we use to reconstruct the input, which in our case, will predict the ratings for movies that the input hasn't watched, which is exactly what we can use to recommend movies!\n",
    "\n",
    "We will now begin to format our dataset to follow the model's expected input."
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "## Formatting the Data"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "First let's see how many movies we have and see if the movie ID's correspond with that value:"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 8,
   "metadata": {
    "collapsed": false
   },
   "outputs": [
    {
     "data": {
      "text/plain": [
       "3883"
      ]
     },
     "execution_count": 8,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "len(movies_df)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 9,
   "metadata": {
    "collapsed": false,
    "scrolled": true
   },
   "outputs": [
    {
     "data": {
      "text/html": [
       "<div>\n",
       "<table border=\"1\" class=\"dataframe\">\n",
       "  <thead>\n",
       "    <tr style=\"text-align: right;\">\n",
       "      <th></th>\n",
       "      <th>MovieID</th>\n",
       "      <th>Title</th>\n",
       "      <th>Genres</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>3878</th>\n",
       "      <td>3948</td>\n",
       "      <td>Meet the Parents (2000)</td>\n",
       "      <td>Comedy</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3879</th>\n",
       "      <td>3949</td>\n",
       "      <td>Requiem for a Dream (2000)</td>\n",
       "      <td>Drama</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3880</th>\n",
       "      <td>3950</td>\n",
       "      <td>Tigerland (2000)</td>\n",
       "      <td>Drama</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3881</th>\n",
       "      <td>3951</td>\n",
       "      <td>Two Family House (2000)</td>\n",
       "      <td>Drama</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3882</th>\n",
       "      <td>3952</td>\n",
       "      <td>Contender, The (2000)</td>\n",
       "      <td>Drama|Thriller</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "      MovieID                       Title          Genres\n",
       "3878     3948     Meet the Parents (2000)          Comedy\n",
       "3879     3949  Requiem for a Dream (2000)           Drama\n",
       "3880     3950            Tigerland (2000)           Drama\n",
       "3881     3951     Two Family House (2000)           Drama\n",
       "3882     3952       Contender, The (2000)  Drama|Thriller"
      ]
     },
     "execution_count": 9,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "movies_df.tail()"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "As it is possible to notice, we have 3883 movies, while our ID's vary from 1 to 3952. Due to this, we won't be able to index movies through their ID since we would get memory indexing errors. To amend this, we can create a column that shows what spot in our list that particular movie is in:"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 10,
   "metadata": {
    "collapsed": false,
    "scrolled": true
   },
   "outputs": [
    {
     "data": {
      "text/html": [
       "<div>\n",
       "<table border=\"1\" class=\"dataframe\">\n",
       "  <thead>\n",
       "    <tr style=\"text-align: right;\">\n",
       "      <th></th>\n",
       "      <th>MovieID</th>\n",
       "      <th>Title</th>\n",
       "      <th>Genres</th>\n",
       "      <th>List Index</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>1</td>\n",
       "      <td>Toy Story (1995)</td>\n",
       "      <td>Animation|Children's|Comedy</td>\n",
       "      <td>0</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>2</td>\n",
       "      <td>Jumanji (1995)</td>\n",
       "      <td>Adventure|Children's|Fantasy</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>3</td>\n",
       "      <td>Grumpier Old Men (1995)</td>\n",
       "      <td>Comedy|Romance</td>\n",
       "      <td>2</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>4</td>\n",
       "      <td>Waiting to Exhale (1995)</td>\n",
       "      <td>Comedy|Drama</td>\n",
       "      <td>3</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>5</td>\n",
       "      <td>Father of the Bride Part II (1995)</td>\n",
       "      <td>Comedy</td>\n",
       "      <td>4</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "   MovieID                               Title                        Genres  \\\n",
       "0        1                    Toy Story (1995)   Animation|Children's|Comedy   \n",
       "1        2                      Jumanji (1995)  Adventure|Children's|Fantasy   \n",
       "2        3             Grumpier Old Men (1995)                Comedy|Romance   \n",
       "3        4            Waiting to Exhale (1995)                  Comedy|Drama   \n",
       "4        5  Father of the Bride Part II (1995)                        Comedy   \n",
       "\n",
       "   List Index  \n",
       "0           0  \n",
       "1           1  \n",
       "2           2  \n",
       "3           3  \n",
       "4           4  "
      ]
     },
     "execution_count": 10,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "movies_df['List Index'] = movies_df.index\n",
    "movies_df.head()"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "With that, let's merge the ratings dataframe into the movies one so we can have the List Index values in both dataframes. Additionally we're also going to drop the Timestamp, Title and Genres columns since we won't be needing it to make recommendations."
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 11,
   "metadata": {
    "collapsed": false,
    "scrolled": false
   },
   "outputs": [
    {
     "data": {
      "text/html": [
       "<div>\n",
       "<table border=\"1\" class=\"dataframe\">\n",
       "  <thead>\n",
       "    <tr style=\"text-align: right;\">\n",
       "      <th></th>\n",
       "      <th>MovieID</th>\n",
       "      <th>List Index</th>\n",
       "      <th>UserID</th>\n",
       "      <th>Rating</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>1</td>\n",
       "      <td>5</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>6</td>\n",
       "      <td>4</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>8</td>\n",
       "      <td>4</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>9</td>\n",
       "      <td>5</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>10</td>\n",
       "      <td>5</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "   MovieID  List Index  UserID  Rating\n",
       "0        1           0       1       5\n",
       "1        1           0       6       4\n",
       "2        1           0       8       4\n",
       "3        1           0       9       5\n",
       "4        1           0      10       5"
      ]
     },
     "execution_count": 11,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "#Merging movies_df with ratings_df by MovieID\n",
    "merged_df = movies_df.merge(ratings_df, on='MovieID')\n",
    "#Dropping unecessary columns\n",
    "merged_df = merged_df.drop('Timestamp', axis=1).drop('Title', axis=1).drop('Genres', axis=1)\n",
    "#Displaying the result\n",
    "merged_df.head()"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "Let's also group up the users by their user IDs and take a look at one of them."
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 12,
   "metadata": {
    "collapsed": false
   },
   "outputs": [
    {
     "data": {
      "text/html": [
       "<div>\n",
       "<table border=\"1\" class=\"dataframe\">\n",
       "  <thead>\n",
       "    <tr style=\"text-align: right;\">\n",
       "      <th></th>\n",
       "      <th>MovieID</th>\n",
       "      <th>List Index</th>\n",
       "      <th>Rating</th>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>UserID</th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>1</td>\n",
       "      <td>0</td>\n",
       "      <td>5</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>21</td>\n",
       "      <td>20</td>\n",
       "      <td>1</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>104</td>\n",
       "      <td>102</td>\n",
       "      <td>4</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>260</td>\n",
       "      <td>257</td>\n",
       "      <td>5</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>5</th>\n",
       "      <td>6</td>\n",
       "      <td>5</td>\n",
       "      <td>2</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "        MovieID  List Index  Rating\n",
       "UserID                             \n",
       "1             1           0       5\n",
       "2            21          20       1\n",
       "3           104         102       4\n",
       "4           260         257       5\n",
       "5             6           5       2"
      ]
     },
     "execution_count": 12,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "#Group up by UserID\n",
    "userGroup = merged_df.groupby('UserID')\n",
    "userGroup.first().head()"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "Now, we can start formatting the data into input for the RBM. We're going to store the normalized users ratings into a list of lists called trX."
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 13,
   "metadata": {
    "collapsed": false
   },
   "outputs": [],
   "source": [
    "#Amount of users used for training\n",
    "amountOfUsedUsers = 1000\n",
    "#Creating the training list\n",
    "trX = []\n",
    "#For each user in the group\n",
    "for userID, curUser in userGroup:\n",
    "    #Create a temp that stores every movie's rating\n",
    "    temp = [0]*len(movies_df)\n",
    "    #For each movie in curUser's movie list\n",
    "    for num, movie in curUser.iterrows():\n",
    "        #Divide the rating by 5 and store it\n",
    "        temp[movie['List Index']] = movie['Rating']/5.0\n",
    "    #Now add the list of ratings into the training list\n",
    "    trX.append(temp)\n",
    "    #Check to see if we finished adding in the amount of users for training\n",
    "    if amountOfUsedUsers == 0:\n",
    "        break\n",
    "    amountOfUsedUsers -= 1"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "------------"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "<a id=\"ref4\"></a>\n",
    "# Setting the Model's Parameters"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "Next, let's start building our RBM with Tensorflow. We'll begin by first determining the amount of hidden layers and then creating placeholder variables for storing our visible layer biases, hidden layer biases and weights that connect the hidden layer with the visible one. We will be arbitrarily setting the amount of hidden layers to 20. You can freely set this value to any number you want since each neuron in the hidden layer will end up learning a feature."
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 14,
   "metadata": {
    "collapsed": false
   },
   "outputs": [],
   "source": [
    "hiddenUnits = 20\n",
    "visibleUnits = len(movies_df)\n",
    "vb = tf.placeholder(\"float\", [visibleUnits]) #Number of unique movies\n",
    "hb = tf.placeholder(\"float\", [hiddenUnits]) #Number of features we're going to learn\n",
    "W = tf.placeholder(\"float\", [visibleUnits, hiddenUnits])"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "We then move on to creating the visible and hidden layer units and setting their activation functions. In this case, we will be using the `tf.sigmoid` and `tf.relu` functions as nonlinear activations since it's what is usually used in RBM's."
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 15,
   "metadata": {
    "collapsed": true
   },
   "outputs": [],
   "source": [
    "#Phase 1: Input Processing\n",
    "v0 = tf.placeholder(\"float\", [None, visibleUnits])\n",
    "_h0= tf.nn.sigmoid(tf.matmul(v0, W) + hb)\n",
    "h0 = tf.nn.relu(tf.sign(_h0 - tf.random_uniform(tf.shape(_h0))))\n",
    "#Phase 2: Reconstruction\n",
    "_v1 = tf.nn.sigmoid(tf.matmul(h0, tf.transpose(W)) + vb) \n",
    "v1 = tf.nn.relu(tf.sign(_v1 - tf.random_uniform(tf.shape(_v1))))\n",
    "h1 = tf.nn.sigmoid(tf.matmul(v1, W) + hb)"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {
    "collapsed": true
   },
   "source": [
    "Now we set the RBM training parameters and functions."
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 16,
   "metadata": {
    "collapsed": false
   },
   "outputs": [],
   "source": [
    "#Learning rate\n",
    "alpha = 1.0\n",
    "#Create the gradients\n",
    "w_pos_grad = tf.matmul(tf.transpose(v0), h0)\n",
    "w_neg_grad = tf.matmul(tf.transpose(v1), h1)\n",
    "#Calculate the Contrastive Divergence to maximize\n",
    "CD = (w_pos_grad - w_neg_grad) / tf.to_float(tf.shape(v0)[0])\n",
    "#Create methods to update the weights and biases\n",
    "update_w = W + alpha * CD\n",
    "update_vb = vb + alpha * tf.reduce_mean(v0 - v1, 0)\n",
    "update_hb = hb + alpha * tf.reduce_mean(h0 - h1, 0)"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "And set the error function, which in this case will be the Mean Absolute Error Function."
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 17,
   "metadata": {
    "collapsed": false
   },
   "outputs": [],
   "source": [
    "err = v0 - v1\n",
    "err_sum = tf.reduce_mean(err * err)"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "We also have to initialize our variables. Thankfully, NumPy has a handy `zeros` function for this. We use it like so:"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 18,
   "metadata": {
    "collapsed": false
   },
   "outputs": [],
   "source": [
    "#Current weight\n",
    "cur_w = np.zeros([visibleUnits, hiddenUnits], np.float32)\n",
    "#Current visible unit biases\n",
    "cur_vb = np.zeros([visibleUnits], np.float32)\n",
    "#Current hidden unit biases\n",
    "cur_hb = np.zeros([hiddenUnits], np.float32)\n",
    "#Previous weight\n",
    "prv_w = np.zeros([visibleUnits, hiddenUnits], np.float32)\n",
    "#Previous visible unit biases\n",
    "prv_vb = np.zeros([visibleUnits], np.float32)\n",
    "#Previous hidden unit biases\n",
    "prv_hb = np.zeros([hiddenUnits], np.float32)\n",
    "sess = tf.Session()\n",
    "sess.run(tf.global_variables_initializer())"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "Now we train the RBM with 15 epochs with each epoch using 10 batches with size 100. After training, we print out a graph with the error by epoch."
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 19,
   "metadata": {
    "collapsed": false,
    "scrolled": false
   },
   "outputs": [
    {
     "data": {
      "image/png": "iVBORw0KGgoAAAANSUhEUgAAAYwAAAEPCAYAAABRHfM8AAAABHNCSVQICAgIfAhkiAAAAAlwSFlz\nAAALEgAACxIB0t1+/AAAIABJREFUeJzt3Xu8VXWd//HXGxBJURRFTAgSUVFMwQtR6LjTSrRGfDRd\noOlhWqOUms1oiV1mPDVTD63pouGkljLZaPjLrJjG/FnqHvOGgGKIIPhTkZsoN5UM5fL5/bHWgc1x\nn3P2Puess/be5/18PPbjrPVd373251icz/5+v+v7/SoiMDMza0+vvAMwM7P64IRhZmYVccIwM7OK\nOGGYmVlFnDDMzKwiThhmZlaRzBOGpImSFktaImlameuHS3pI0mZJl5SU7y5ptqTHJS2U9O2sYzUz\ns9Ypy3kYknoBS4BTgVXAHGByRCwuqbM/MBw4C9gQEd8vubZHRLwuqTfwIHBpRDyYWcBmZtaqrFsY\n44ClEbEsIrYAM4FJpRUiYm1EzAO2tnxzRLyeHu6exroh43jNzKwVWSeMIcDykvMVaVlFJPWS9Djw\nIlCMiKe6OD4zM6tQTQ96R8T2iBgLDAX+RtLJecdkZtZT9cn4/iuBYSXnQ9OyqkTEq5L+Bzge+N+W\n1yV5QSwzsypFhKqpn3ULYw4wUtJwSX2BycCsNurvCF7S/pIGpMdvAz4AzG/tjRFRl68rrrgi9xgc\nf/5xOP76fNVz/B2RaQsjIrZJugi4myQ53RgRiyRNTS7HDZIGA3OBvYDtkr4IHAm8HfiZJKXv/XlE\n3JNlvGZm1rqsu6SIiLuAw1uUXV9yvAZ4R5m3LgCOzTY6MzOrVE0PevcEhUIh7xA6xfHny/Hnq97j\nr1amE/e6i6RohN/DzKy7SCJqbNDbzMwahBOGmZlVxAnDzMwq4oRhZmYVccIwM7OKOGGYmVlFnDDM\nzKwiThhmZlYRJwwzM6uIE4aZmVXECcPMzCrihGFmZhVxwjAzs4o0TMLYuDHvCMzMGlvmCUPSREmL\nJS2RNK3M9cMlPSRps6RLSsqHSrpX0kJJCyRd3NbnPPRQFtGbmVmzTBOGpF7AdOA0YDQwRdKoFtXW\nAV8AvtuifCtwSUSMBt4DXFjmvTv86U9dFraZmZWRdQtjHLA0IpZFxBZgJjCptEJErI2IeSQJorT8\nxYiYnx5vAhYBQ1r7ICcMM7NsZZ0whgDLS85X0MYf/dZIeicwBpjdWp3582Hz5mrvbGZmleqTdwDt\nkdQfuB34YtrSKGvAgCY+/3kYPjzZZ7en7bVrZtaWYrFIsVjs1D0y3dNb0nigKSImpueXAxERV5Wp\newXwWkR8v6SsD/A74PcRcXUbnxOXXBIMHAhf+1qX/xpmZg2nFvf0ngOMlDRcUl9gMjCrjfotg78J\neKqtZNHspJM8jmFmlqVMWxiQPFYLXE2SnG6MiCslTSVpadwgaTAwF9gL2A5sAo4EjgHuBxYAkb6+\nGhF3lfmMeOmlYORIWL8eevfO9FcyM6t7HWlhZJ4wuoOkiAiOOAJuvRXGjs07IjOz2laLXVLdyt1S\nZmbZccIwM7OKNFSX1PPPw/jxsHo1qKqGlplZz9Lju6SGD4fddoNnnsk7EjOzxtNQCUNyt5SZWVYa\nKmGAE4aZWVYaLmGceKIThplZFhouYYwenUzeW70670jMzBpLwyWMXr1gwgR44IG8IzEzaywNlzDA\n4xhmZllwwjAzs4o01MS9Zm++CQMHwsqVMGBAjoGZmdWoHj9xr1nfvnDCCfDQQ3lHYmbWOBoyYYC7\npczMulrDJgzPxzAz61oNOYYB8NprcOCBsG4d9OuXU2BmZjWqJscwJE2UtFjSEknTylw/XNJDkjZL\nuqTFtRslrZH052o/d6+94IgjYO7czkRvZmbNMk0YknoB04HTgNHAFEmjWlRbB3wB+G6ZW8xI39sh\nHscwM+s6WbcwxgFLI2JZRGwBZgKTSitExNqImAdsbfnmiHgA2NDRD3fCMDPrOlknjCHA8pLzFWlZ\ntzjxxOTR2m3buusTzcwaV5+8A+gqTU1NO44LhQKFQoEDDkgGvhcsgDFj8ovNzCxvxWKRYrHYqXtk\n+pSUpPFAU0RMTM8vByIiripT9wrgtYj4fovy4cB/R8TRbXzOW56SanbeeXD00fCFL3TiFzEzazC1\n+JTUHGCkpOGS+gKTgVlt1C8XvFopr4jnY5iZdY3M52FImghcTZKcboyIKyVNJWlp3CBpMDAX2AvY\nDmwCjoyITZJuBQrAfsAa4IqImFHmM1ptYTz7bJI0Vq5MtnA1M7OOtTAaduJeswgYOhTuvx8OOaSb\nAzMzq1G12CWVO8mP15qZdYWGTxjghGFm1hWcMMzMrCI9ImEcdRS8/DK8+GLekZiZ1a8ekTB69YIJ\nE+CBB/KOxMysfvWIhAGej2Fm1lk9JmGcdJJbGGZmndHw8zCavfEG7LcfrFoFe+/dTYGZmdUoz8No\nw+67w3HHwcMP5x2JmVl96jEJA/x4rZlZZzhhmJlZRXrMGAbAq6/CQQfBunVJF5WZWU/lMYx27L03\nHH44zJ2bdyRmZvWnRyUM8HwMM7OO6nEJw/MxzMw6pkeNYQCsWQOjRiXjGL16XLo0M0vU5BiGpImS\nFktaImlameuHS3pI0mZJl1Tz3o4YPBgGDYInn+yKu5mZ9RyZJgxJvYDpwGnAaGCKpFEtqq0DvgB8\ntwPv7RA/XmtmVr2sWxjjgKURsSwitgAzgUmlFSJibUTMA7ZW+96OcsIwM6te1gljCLC85HxFWpb1\ne9vUnDAaYPjGzKzb9Mk7gK7S1NS047hQKFAoFFqtO2JEkiyeey45NjNrdMVikWKx2Kl7ZJ0wVgLD\nSs6HpmVd/t7ShNEeaed8DCcMM+sJWn6R/sY3vlH1PbLukpoDjJQ0XFJfYDIwq436pY94Vfveqng+\nhplZdTJtYUTENkkXAXeTJKcbI2KRpKnJ5bhB0mBgLrAXsF3SF4EjI2JTufd2VWwnnQTXXttVdzMz\na3w9buJes23bkg2VliyBAw7IKDAzsxpVkxP3alXv3vDe97pbysysUj02YYDnY5iZVcMJwwnDzKwi\nPXYMA2Dz5mQc48UXYa+9MgjMzKxGeQyjSv36wbHHwiOP5B2JmVnt69EJA9wtZWZWKScMJwwzs4r0\n6DEMgFdegSFDYP166Nu3iwMzM6tRHsPogAED4NBDYd68vCMxM6ttPT5hgLulzMwq4YSBE4aZWSV6\n/BgGwOrVMHo0rF0LvZxCzawH8BhGB7397TBwIDz1VN6RmJnVLieMlLulzMza5oSRcsIwM2tb5glD\n0kRJiyUtkTStlTrXSFoqab6kMSXlX5S0IH1dnGWczQmjAYZ0zMwykWnCkNQLmA6cBowGpkga1aLO\n6cAhEXEoMBW4Li0fDXwWOB4YA3xYUmY7cI8cCVu2wLJlWX2CmVl9y7qFMQ5YGhHLImILMBOY1KLO\nJOBmgIiYDQxIt209ApgdEW9ExDbgfuAjWQUquVvKzKwtWSeMIcDykvMVaVlbdVamZU8CJ0naV9Ie\nwBnAOzKM1QnDzKwNNTvoHRGLgauAPwB3Ao8D27L8zBNPdMIwM2tNn4zvvxIYVnI+NC1rWecd5epE\nxAxgBoCkb7FrS2QXTU1NO44LhQKFQqHqYI85BlatgpdfhkGDqn67mVnNKhaLFIvFTt0j05neknoD\nTwOnAquBR4EpEbGopM4ZwIUR8SFJ44EfRsT49NqgiHhZ0jDgLmB8RLxa5nM6NdO71MSJ8LnPwVln\ndcntzMxqUiYzvSX1lvTvHQkoHay+CLgbWAjMjIhFkqZKOj+tcyfwnKRngOuBC0pu8StJTwK/BS4o\nlyy6mscxzMzKq6iFIemR5m/9tagrWxj33w9f+hI8+miX3M7MrCZ1pIVRacL4McmTS78E/tJcHhF3\nVBtkFroyYWzeDPvvDy++CP37d8ktzcxqTkcSRqWD3v2AdcApJWUB1ETC6Er9+sGYMfDII/D+9+cd\njZlZ7agoYUTEuVkHUkuaxzGcMMzMdqpoHoakoZJ+Leml9PUrSUOzDi4vno9hZvZWlU7cmwHMAg5K\nX/+dljWkCRNgzpxkbSkzM0tUmjAGRcSMiNiavv4TaNipbfvsAyNGwGOP5R2JmVntqDRhrJP0qXRO\nRm9JnyIZBG9Yno9hZrarShPGZ4CPAy+SzNj+KNDQA+FOGGZmu2p3Hka6vMfFEfGD7gmpel05D6PZ\nqlXwrncl60r1qtklGs3MOiaTpUHS5T2mdDiqOnXQQclYxqJF7dc1M+sJKv3u/KCk6ZJOknRs8yvT\nyGqAu6XMzHaqdGmQ+8oUR0ScUqa822XRJQXw05/CfffBLbd0+a3NzHKVyVpS6b7cH42I/9OZ4LKU\nVcJ4+ulktveyZR7HMLPGktUYxnbgsg5HVccOOywZy/jJT/KOxMwsf5V2SV0JrAVuY9fVatdnF1rl\nsmphACxcCIUCzJsHw4a1W93MrC5kubz5c2WKIyJGVPNhWckyYQB861vJPhl33QWq6j+vmVltyqRL\nCiAiDi7zqihZSJooabGkJZKmtVLnGklLJc2XNKak/CuSFkr6s6RbJPWt7NfqWpddlszH+M//zOPT\nzcxqQ5sJQ9JlJccfa3Ht2+3dPB0wnw6cBowGpkga1aLO6cAhEXEoMBW4Li0fDpwHjI2Io0mWYp9c\nwe/U5XbbDWbMgGnTYOXKPCIwM8tfey2M0j/QX2lxbWIF9x8HLI2IZRGxBZgJTGpRZxJwM0BEzAYG\nSBoMvAq8CewpqQ+wB7Cqgs/MxDHHwOc/D5/7HGTY+2VmVrPaSxhq5bjceTlDgOUl5yvSsrbqrASG\nRMQG4HvAC2nZxoj4YwWfmZmvfQ2efx5uvTXPKMzM8tFewohWjsuddylJI4B/AoaT7MHRX9Ins/zM\n9vTtm3RNXXJJsue3mVlP0t4WrcdIepWkNfG29Jj0vF8F918JlD6MOjQta1nnHWXqnAw82PzorqQ7\ngPcCZb/fNzU17TguFAoUCoUKwqve8cfDZz4DF14It9/up6bMrD4Ui0WKxWKn7lHRY7Udvnmy0u3T\nwKkky6I/CkyJiEUldc4ALoyID0kaD/wwIsZLOgb4L+AE4A2SHf7mRMS1ZT4n08dqW9q8GcaOhW9+\nEz72sfbrm5nVmo48VtteC6NTImKbpIuAu0m6v26MiEWSpiaX44aIuFPSGZKeIZkUeG763ick3QzM\nA7YBjwM3ZBlvpfr1g5tugo98JJnUN6hh9x40M9sp0xZGd+nuFkazSy9N9s34xS+6/aPNzDols4l7\nVt6//ivMnQu/+U3ekZiZZc8tjE66/36YMgUWLICBA3MJwcysapmtJVXr8kwYABddBK+9Bj/7WW4h\nmJlVxQkjJ5s2Jft/X3stnHFGbmGYmVXMYxg56d8/2Z1v6lR45ZW8ozEzy4ZbGF3o/POTnzfUxMO/\nZmatc5dUzl55JemauummZGtXM7Na5S6pnA0YANdfD+edlwyCm5k1ErcwMnDOOcm4xvTpeUdiZlae\nu6RqxIYNcNRRyTLoJ5+cdzRmZm/lLqkase++8OMfw2c/C6+/nnc0ZmZdwy2MDH3yk3DggfD97+cd\niZnZrtwlVWPWrk2emvrVr+C97807GjOzndwlVWP23x9+9KNkw6W//jXvaMzMOsctjG7w0Y/CyJFw\n5ZV5R2JmlnCXVI1aswaOPhp+9zs44YS8ozEzq9EuKUkTJS2WtETStFbqXCNpqaT5ksakZYdJelzS\nY+nPVyRdnHW8WRg8GH7wAzj3XHjjjbyjMTPrmKz39O4FLCHZ03sVMAeYHBGLS+qcDlyU7un9buDq\niBhf5j4rgHdHxPIyn1PTLQyACJg0CcaMSfYCNzPLUy22MMYBSyNiWURsAWYCk1rUmQTcDBARs4EB\nkga3qPN+4P+VSxb1QoLrrkte8+fnHY2ZWfWyThhDgNI/8ivSsrbqrCxT5xNA3e+cfdBBcNVVSdfU\nli15R2NmVp0+eQfQHkm7AWcCl7dVr6mpacdxoVCgUChkGldHnXMO3HZbkji+/vW8ozGznqJYLFIs\nFjt1j6zHMMYDTRExMT2/HIiIuKqkznXAfRFxW3q+GDg5Itak52cCFzTfo5XPqfkxjFIvvADHHQf3\n3ZesOWVm1t1qcQxjDjBS0nBJfYHJwKwWdWYBZ8OOBLOxOVmkptAA3VGlhg2Df/u3ZELf1q15R2Nm\nVplME0ZEbAMuAu4GFgIzI2KRpKmSzk/r3Ak8J+kZ4Hrggub3S9qDZMD7jizjzMP558Nee3mdKTOr\nH564l6PnnoP3vCcZy7jwwuRJKjOz7uCZ3nXo2WfhzDNhwoRk3am+ffOOyMx6glocw7B2jBgBDz0E\nq1fDBz+YrHBrZlaLnDBqwN57w69/nXRPjRsHTz6Zd0RmZm/lLqka81//BZdcAjfeCH/7t3lHY2aN\nqiNdUjU/ca+n+dSn4NBD4SMfgaeegssu82C4mdUGtzBq1IoVyWKFRx4JP/kJ9OuXd0Rm1kg86N1A\nhg6FP/0J3nwTCoVkUNzMLE9OGDVsjz1g5kz48IeTwfB58/KOyMx6MndJ1Yk77oCpU+Haa+HjH887\nGjOrd5641+Dmz4ezzoKzz4amJujl9qGZdZATRg+wZk3yBNWBB8LNN8Oee+YdkZnVIw969wCDB8O9\n98KAAclyIsuW5R2RmfUUThh1aPfdk4l9Z5+dzA5/8MG8IzKznsBdUnXu97+HT38avvOdZDc/M7NK\neAyjh1q0KFnxdtKkZOvX3r3zjsjMap0TRg+2fn3yuG3fvvCLXyRjHGZmranJQW9JEyUtlrRE0rRW\n6lwjaamk+ZLGlJQPkPRLSYskLZT07qzjrVcDBybdUyNGwPjxsHRp3hGZWaPJNGFI6gVMB04DRgNT\nJI1qUed04JCIOBSYClxXcvlq4M6IOAI4BliUZbz1brfdYPp0+Md/hBNPhHvuyTsiM2skWbcwxgFL\nI2JZRGwBZgKTWtSZBNwMEBGzgQGSBkvaGzgpImak17ZGxKsZx9sQpk6F226Dv//7ZBe/Ht5bZ2Zd\nJOuEMQRYXnK+Ii1rq87KtOxgYK2kGZIek3SDpLdlGm0DKRSSnfxuuikZ29i4Me+IzKze1fJ+GH2A\nY4ELI2KupB8ClwNXlKvc1NS047hQKFAoFLohxNo2YgQ8/DB8+cswdmwyGD5+fN5RmVkeisUixWKx\nU/fI9CkpSeOBpoiYmJ5fDkREXFVS5zrgvoi4LT1fDJycXn44Ikak5ScC0yLiLfvQ+Smp9v3mN0lX\n1aWXwpe+5HWozHq6WnxKag4wUtJwSX2BycCsFnVmAWfDjgSzMSLWRMQaYLmkw9J6pwJPZRxvwzrr\nLJgzB2bNgjPOgJdeyjsiM6s3mSaMiNgGXATcDSwEZkbEIklTJZ2f1rkTeE7SM8D1wAUlt7gYuEXS\nfJKnpL6dZbyNbtgwKBbh+OOTLio/RWVm1fDEvR7qnnuStajOPTdZKr1PLY9mmVmX80xvq8qaNck6\nVJs2wa23Ji0QM+sZanEMw2rY4MFw553JOlQnnJAMjJuZtcYtDAPgkUdgypRk//Dvfhf69cs7IjPL\nklsY1mHjx8Pjj8Pq1ckeG0uW5B2RmdUaJwzbYZ994Je/hM99LtnN7+c/zzsiM6sl7pKyshYsgE98\nIhnbuPZa6N8/74jMrCu5S8q6zLvelUz069MHjjsO5s/POyIzy5sThrVqzz2TvcOvuAI+8IFk6XQ3\n5Mx6LndJWUWeeSbpoho2LEkiAwfmHZGZdYa7pCwzI0cmy6UffDAceyw8+GDeEZlZd3MLw6r2u9/B\nP/wDXHwxTJsGvXvnHZGZVctLg1i3WbEi2dFv+3b4zneSuRtmVj/cJWXdZujQZAHDT386Gdv48If9\nJJVZo3PCsA7r0yfpmlq6FD74QTj99GQ72MWL847MzLLghGGdtvvuyXjGM88kA+InnQTnnAPPPZd3\nZGbWlZwwrMvsuSdcfnnS4hg2LNmo6YILYNWqvCMzs66QecKQNFHSYklLJE1rpc41kpZKmi9pbEn5\n85KekPS4pEezjtW6xj77wDe/mXRN7bEHHHVUso/42rV5R2ZmnZFpwpDUC5gOnAaMBqZIGtWizunA\nIRFxKDAV+HHJ5e1AISLGRsS4LGO1rjdoEPz7v8OTT8Lrr8Phh8O//Au88krekZlZR2TdwhgHLI2I\nZRGxBZgJTGpRZxJwM0BEzAYGSBqcXlM3xGgZO+gg+I//gLlzYflyOPRQuPJK+Mtf8o7MzKqR9R/j\nIcDykvMVaVlbdVaW1AngD5LmSDovsyitWxx8MMyYAfffn+y9MXIkXHMNvPFG3pGZWSX65B1AOyZE\nxGpJg0gSx6KIeKBcxaamph3HhUKBQqHQPRFa1UaNgttuS+Zt/PM/w/e+l/w855zkUV0z63rFYpFi\nsdipe2Q601vSeKApIiam55cDERFXldS5DrgvIm5LzxcDJ0fEmhb3ugJ4LSK+X+ZzPNO7jj38MHz9\n6/DCC/CNb8DkydDLHZFmmarFmd5zgJGShkvqC0wGZrWoMws4G3YkmI0RsUbSHpL6p+V7Ah8Ensw4\nXsvBe96TzBq//nr40Y/gmGPgN7/xUupmtSbztaQkTQSuJklON0bElZKmkrQ0bkjrTAcmAn8Bzo2I\nxyQdDPyaZByjD3BLRFzZyme4hdEgIuB//idpcfTtC1/+Mpx6qpdTN+tqXnzQGsb27XD77XDTTclS\n6ocdliSOU06BE0/0lrFmneWEYQ3pzTfh0UeTbqt774V582Ds2CR5nHIKjB+fLE9iZpVzwrAe4fXX\n4YEHkuRx772waFEyDnLKKUkr5NhjvUeHWXucMKxH2rgR/vd/dyaQFSvgb/5mZxfW6NGgqv5ZmDU+\nJwwzYM0auO++JHnccw9s2gTve9/OBDJihBOImROGWRnPP79rAtltt53JY8IEeOc7nUCs53HCMGtH\nBDz99M7kMXs2vPoqvOtdyevoo3ce77NP3tGaZccJw6wD1q+HBQuS15//nLwWLoR9992ZQJp/Hn54\n0kIxq3dOGGZdZPv2ZMfA5iTS/LN5td2jj941mbz97e7WsvrihGGWsddfh6ee2jWRPPFEkmBatkaO\nOirZhdCsFjlhmOUgInkyq2VrZPFi2GuvZCxk332Tn5Ue77OPu74sW04YZjVk69ZkW9qNG5PXhg3V\nHffr135i2X9/OOAAGDw4+TloULIGl1l7nDDMGkREsiNhW4llwwZYty5p3bz0UvJ6+eWkVXPAAbsm\nktaO997bYy89lROGWQ+3fXuSSF56addE0vK8+fjNN3cmkXJJZeDAt7Zq+vVzkmkEThhmVpW//rXt\npLJ+/c5WTXPLJuKt4y3lxmDKnQ8Y4C6zWuGEYWaZ27y5fDdZJecbNyYJozmJ9O+fPEnWFS8vOFmd\nmkwY6QZKP2TnBkpXlalzDXA6yQZK50TE/JJrvYC5wIqIOLOVz3DCMKsDzWMzzclk06bkvLOv119P\nElHLJNKckNr72da1Ru2C60jC6JNVMLDjj/104FRgFTBH0m8jYnFJndOBQyLiUEnvBq4Dxpfc5ovA\nU8DeWcaal2KxSKFQyDuMDnP8+aq3+KXkD3H//jB0aBL/+99f6PR9I5LutdYSSnNiKk1QL73U+rXS\nsi1byieX/v3hzTeLHHlkgYEDYb/9kjGf0uP99ktaUo3S+sk0YQDjgKURsQxA0kxgErC4pM4k4GaA\niJgtaYCkwem+3kOBM4BvAZdkHGsu6u0ffEuOP1+OPyHBHnskr0GDOh9Xqa1byyef116Dn/60yKhR\nBdatg2efhTlzknGfdeuSn+vXwyuvJE+ulSaRcomlZdk++0CvXl37u3RW1gljCLC85HwFSRJpq87K\ntGwN8APgy8CADGM0M2tVnz7JYP2AMn+FZs+Giy5q+/3btiVJozmJlCaTdetg6VJ45JFdr61blyyK\n2bdvsptkv37Jz9LjzpZ16L9Fx96WPUkfAtZExHxJBaABexHNrNH17r2z9VCN7dvhjTd2vjZv3vVn\ne2XNx5s2JRNIW17viEwHvSWNB5oiYmJ6fjkQpQPfkq4D7ouI29LzxcDJJGMXnwK2Am8D9gLuiIiz\ny3yOR7zNzKpUU09JSeoNPE0y6L0aeBSYEhGLSuqcAVwYER9KE8wPI2J8i/ucDFza2lNSZmaWvUy7\npCJim6SLgLvZ+VjtIklTk8txQ0TcKekMSc+QPFZ7bpYxmZlZxzTExD0zM8tejT20VR1JEyUtlrRE\n0rS846mGpKGS7pW0UNICSRfnHVO1JPWS9JikWXnH0hHpI9y/lLQo/d/h3XnHVClJX0lj/rOkWyTV\n9IIbkm6UtEbSn0vK9pV0t6SnJf1fSTX7NGQr8X8n/f/OfEm/klSzc8XKxV9y7VJJ2yW1Oyxftwmj\nZFLgacBoYIqkUflGVZWtwCURMRp4D3BhncUPOydV1qurgTsj4gjgGGBRO/VrgqThwHnA2Ig4mqRr\neXK+UbVrBsm/1VKXA3+MiMOBe4GvdHtUlSsX/93A6IgYAyyl/uInnev2AWBZJTep24RByaTAiNgC\nNE8KrAsR8WLzEigRsYnkj9WQfKOqXMmkyp/mHUtHpN8GT4qIGQARsTUiXs05rEq9CrwJ7CmpD7AH\nyUoKNSsiHgA2tCieBPwsPf4ZcFa3BlWFcvFHxB8jYnt6+ggwtNsDq1Ar//1h51y3itRzwig3KbBu\n/uCWkvROYAwwO99IqtL8f7R6HQQ7GFgraUbarXaDpLflHVQlImID8D3gBZKJrhsj4o/5RtUhB0TE\nGki+QAEH5BxPZ3wG+H3eQVRD0pnA8ohYUOl76jlhNARJ/YHbgS+mLY2aVzqpkmRCZT1OquwDHAtc\nGxHHAq+TdJHUPEkjgH8ChgMHAf0lfTLfqLpEXX75kPQ1YEtE3Jp3LJVKvxx9FbiitLi999VzwlgJ\nDCs5H5qW1Y20O+F24OcR8du846nCBOBMSc8CvwDeJ+nmnGOq1gqSb1dz0/PbSRJIPTgeeDAi1kfE\nNuAO4L05x9QRayQNBpB0IPBSzvFUTdI5JF2z9ZawDwHeCTwh6TmSv5/zJLXZyqvnhDEHGClpePqE\nyGSg3p7kyMAZAAACpUlEQVTWuQl4KiKuzjuQakTEVyNiWESMIPnvfm+5Gfi1LO0KWS7psLToVOpn\nAP9pYLykfpJEEns9DNi3bI3OAs5Jjz8N1PqXpl3iT7du+DJwZkR0cLGNbrUj/oh4MiIOjIgREXEw\nyReosRHRZtKu24SRfrNqnhS4EJhZOoO81kmaAPw9cIqkx9N+9Il5x9XDXAzcImk+yVNS3845nopE\nxBMkKzzPA54g+SNwQ65BtUPSrcBDwGGSXpB0LnAl8AFJzatBXJlnjG1pJf4fAf2BP6T/fv8j1yDb\n0Er8pYIKuqQ8cc/MzCpSty0MMzPrXk4YZmZWEScMMzOriBOGmZlVxAnDzMwq4oRhZmYVccIwq4Ck\nbemz9s1zZi7rwnsPl1Txej5mecl0xz2zBvKXdM2prHhClNU8tzDMKlN2Fqyk5yRdlW5k9Ei6MGBz\nq+GedHOdP6TLwSPpAEl3pOWPp/vYA/RJV8x9UtJdknbvpt/LrGJOGGaVeVuLLqmPlVzbkG5kdC3J\npkyQLBsxI91c59b0HOAaoJiWH0uyrA3AocCPIuIo4BXg7zL+fcyq5qVBzCog6dWIeMsWnOlKn++L\niOfT1YdXR8QgSS8DB0bEtrR8VUQcIOklYEi66VfzPYYDd6c7z5GOj/SJiLpY28p6DrcwzDovWjmu\nRulqp9vw+KLVICcMs8q0tZLnJ9Kfk4GH0+MHgSnp8aeAP6XHfwQugGRf+nSr2Pbub1YT/C3GrDL9\nJD1G8oc9gLsi4qvptX0lPQFsZmeSuBiYIelLwMtA83LS/wjcIOmzwFbg88CL+CkpqwMewzDrhHQM\n47iIWJ93LGZZc5eUWef4G5f1GG5hmJlZRdzCMDOzijhhmJlZRZwwzMysIk4YZmZWEScMMzOriBOG\nmZlV5P8D9NvX9ahYIiQAAAAASUVORK5CYII=\n",
      "text/plain": [
       "<matplotlib.figure.Figure at 0x7ff0557a1350>"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    }
   ],
   "source": [
    "epochs = 15\n",
    "batchsize = 100\n",
    "errors = []\n",
    "for i in range(epochs):\n",
    "    for start, end in zip( range(0, len(trX), batchsize), range(batchsize, len(trX), batchsize)):\n",
    "        batch = trX[start:end]\n",
    "        cur_w = sess.run(update_w, feed_dict={v0: batch, W: prv_w, vb: prv_vb, hb: prv_hb})\n",
    "        cur_vb = sess.run(update_vb, feed_dict={v0: batch, W: prv_w, vb: prv_vb, hb: prv_hb})\n",
    "        cur_nb = sess.run(update_hb, feed_dict={v0: batch, W: prv_w, vb: prv_vb, hb: prv_hb})\n",
    "        prv_w = cur_w\n",
    "        prv_vb = cur_vb\n",
    "        prv_hb = cur_nb\n",
    "    errors.append(sess.run(err_sum, feed_dict={v0: trX, W: cur_w, vb: cur_vb, hb: cur_nb}))\n",
    "plt.plot(errors)\n",
    "plt.ylabel('Error')\n",
    "plt.xlabel('Epoch')\n",
    "plt.show()"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "--------------"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "<a id=\"ref1337\"></a>\n",
    "## Recommendation"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "We can now predict movies that an arbitrarily selected user might like. This can be accomplished by feeding in the user's watched movie preferences into the RBM and then reconstructing the input. The values that the RBM gives us will attempt to estimate the user's preferences for movies that he hasn't watched based on the preferences of the users that the RBM was trained on."
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 20,
   "metadata": {
    "collapsed": true
   },
   "outputs": [],
   "source": [
    "#Selecting the input user\n",
    "inputUser = [trX[75]]"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 21,
   "metadata": {
    "collapsed": false
   },
   "outputs": [],
   "source": [
    "#Feeding in the user and reconstructing the input\n",
    "hh0 = tf.nn.sigmoid(tf.matmul(v0, W) + hb)\n",
    "vv1 = tf.nn.sigmoid(tf.matmul(hh0, tf.transpose(W)) + vb)\n",
    "feed = sess.run(hh0, feed_dict={ v0: inputUser, W: prv_w, hb: prv_hb})\n",
    "rec = sess.run(vv1, feed_dict={ hh0: feed, W: prv_w, vb: prv_vb})"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "We can then list the 20 most recommended movies for our mock user by sorting it by their scores given by our model."
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 22,
   "metadata": {
    "collapsed": false
   },
   "outputs": [
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "/usr/local/lib/python2.7/dist-packages/ipykernel/__main__.py:2: FutureWarning: sort(columns=....) is deprecated, use sort_values(by=.....)\n",
      "  from ipykernel import kernelapp as app\n"
     ]
    },
    {
     "data": {
      "text/html": [
       "<div>\n",
       "<table border=\"1\" class=\"dataframe\">\n",
       "  <thead>\n",
       "    <tr style=\"text-align: right;\">\n",
       "      <th></th>\n",
       "      <th>MovieID</th>\n",
       "      <th>Title</th>\n",
       "      <th>Genres</th>\n",
       "      <th>List Index</th>\n",
       "      <th>Recommendation Score</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>2789</th>\n",
       "      <td>2858</td>\n",
       "      <td>American Beauty (1999)</td>\n",
       "      <td>Comedy|Drama</td>\n",
       "      <td>2789</td>\n",
       "      <td>0.774460</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2327</th>\n",
       "      <td>2396</td>\n",
       "      <td>Shakespeare in Love (1998)</td>\n",
       "      <td>Comedy|Romance</td>\n",
       "      <td>2327</td>\n",
       "      <td>0.739629</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>315</th>\n",
       "      <td>318</td>\n",
       "      <td>Shawshank Redemption, The (1994)</td>\n",
       "      <td>Drama</td>\n",
       "      <td>315</td>\n",
       "      <td>0.686620</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2928</th>\n",
       "      <td>2997</td>\n",
       "      <td>Being John Malkovich (1999)</td>\n",
       "      <td>Comedy</td>\n",
       "      <td>2928</td>\n",
       "      <td>0.615067</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>589</th>\n",
       "      <td>593</td>\n",
       "      <td>Silence of the Lambs, The (1991)</td>\n",
       "      <td>Drama|Thriller</td>\n",
       "      <td>589</td>\n",
       "      <td>0.610467</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1959</th>\n",
       "      <td>2028</td>\n",
       "      <td>Saving Private Ryan (1998)</td>\n",
       "      <td>Action|Drama|War</td>\n",
       "      <td>1959</td>\n",
       "      <td>0.574823</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2693</th>\n",
       "      <td>2762</td>\n",
       "      <td>Sixth Sense, The (1999)</td>\n",
       "      <td>Thriller</td>\n",
       "      <td>2693</td>\n",
       "      <td>0.574116</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1575</th>\n",
       "      <td>1617</td>\n",
       "      <td>L.A. Confidential (1997)</td>\n",
       "      <td>Crime|Film-Noir|Mystery|Thriller</td>\n",
       "      <td>1575</td>\n",
       "      <td>0.556403</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>523</th>\n",
       "      <td>527</td>\n",
       "      <td>Schindler's List (1993)</td>\n",
       "      <td>Drama|War</td>\n",
       "      <td>523</td>\n",
       "      <td>0.525403</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>604</th>\n",
       "      <td>608</td>\n",
       "      <td>Fargo (1996)</td>\n",
       "      <td>Crime|Drama|Thriller</td>\n",
       "      <td>604</td>\n",
       "      <td>0.517156</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1245</th>\n",
       "      <td>1265</td>\n",
       "      <td>Groundhog Day (1993)</td>\n",
       "      <td>Comedy|Romance</td>\n",
       "      <td>1245</td>\n",
       "      <td>0.512652</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>352</th>\n",
       "      <td>356</td>\n",
       "      <td>Forrest Gump (1994)</td>\n",
       "      <td>Comedy|Romance|War</td>\n",
       "      <td>352</td>\n",
       "      <td>0.481924</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1656</th>\n",
       "      <td>1704</td>\n",
       "      <td>Good Will Hunting (1997)</td>\n",
       "      <td>Drama</td>\n",
       "      <td>1656</td>\n",
       "      <td>0.444412</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>293</th>\n",
       "      <td>296</td>\n",
       "      <td>Pulp Fiction (1994)</td>\n",
       "      <td>Crime|Drama</td>\n",
       "      <td>293</td>\n",
       "      <td>0.408996</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>1</td>\n",
       "      <td>Toy Story (1995)</td>\n",
       "      <td>Animation|Children's|Comedy</td>\n",
       "      <td>0</td>\n",
       "      <td>0.405530</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1195</th>\n",
       "      <td>1213</td>\n",
       "      <td>GoodFellas (1990)</td>\n",
       "      <td>Crime|Drama</td>\n",
       "      <td>1195</td>\n",
       "      <td>0.397025</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>220</th>\n",
       "      <td>223</td>\n",
       "      <td>Clerks (1994)</td>\n",
       "      <td>Comedy</td>\n",
       "      <td>220</td>\n",
       "      <td>0.384806</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3682</th>\n",
       "      <td>3751</td>\n",
       "      <td>Chicken Run (2000)</td>\n",
       "      <td>Animation|Children's|Comedy</td>\n",
       "      <td>3682</td>\n",
       "      <td>0.379062</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2286</th>\n",
       "      <td>2355</td>\n",
       "      <td>Bug's Life, A (1998)</td>\n",
       "      <td>Animation|Children's|Comedy</td>\n",
       "      <td>2286</td>\n",
       "      <td>0.356733</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3509</th>\n",
       "      <td>3578</td>\n",
       "      <td>Gladiator (2000)</td>\n",
       "      <td>Action|Drama</td>\n",
       "      <td>3509</td>\n",
       "      <td>0.353293</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "      MovieID                             Title  \\\n",
       "2789     2858            American Beauty (1999)   \n",
       "2327     2396        Shakespeare in Love (1998)   \n",
       "315       318  Shawshank Redemption, The (1994)   \n",
       "2928     2997       Being John Malkovich (1999)   \n",
       "589       593  Silence of the Lambs, The (1991)   \n",
       "1959     2028        Saving Private Ryan (1998)   \n",
       "2693     2762           Sixth Sense, The (1999)   \n",
       "1575     1617          L.A. Confidential (1997)   \n",
       "523       527           Schindler's List (1993)   \n",
       "604       608                      Fargo (1996)   \n",
       "1245     1265              Groundhog Day (1993)   \n",
       "352       356               Forrest Gump (1994)   \n",
       "1656     1704          Good Will Hunting (1997)   \n",
       "293       296               Pulp Fiction (1994)   \n",
       "0           1                  Toy Story (1995)   \n",
       "1195     1213                 GoodFellas (1990)   \n",
       "220       223                     Clerks (1994)   \n",
       "3682     3751                Chicken Run (2000)   \n",
       "2286     2355              Bug's Life, A (1998)   \n",
       "3509     3578                  Gladiator (2000)   \n",
       "\n",
       "                                Genres  List Index  Recommendation Score  \n",
       "2789                      Comedy|Drama        2789              0.774460  \n",
       "2327                    Comedy|Romance        2327              0.739629  \n",
       "315                              Drama         315              0.686620  \n",
       "2928                            Comedy        2928              0.615067  \n",
       "589                     Drama|Thriller         589              0.610467  \n",
       "1959                  Action|Drama|War        1959              0.574823  \n",
       "2693                          Thriller        2693              0.574116  \n",
       "1575  Crime|Film-Noir|Mystery|Thriller        1575              0.556403  \n",
       "523                          Drama|War         523              0.525403  \n",
       "604               Crime|Drama|Thriller         604              0.517156  \n",
       "1245                    Comedy|Romance        1245              0.512652  \n",
       "352                 Comedy|Romance|War         352              0.481924  \n",
       "1656                             Drama        1656              0.444412  \n",
       "293                        Crime|Drama         293              0.408996  \n",
       "0          Animation|Children's|Comedy           0              0.405530  \n",
       "1195                       Crime|Drama        1195              0.397025  \n",
       "220                             Comedy         220              0.384806  \n",
       "3682       Animation|Children's|Comedy        3682              0.379062  \n",
       "2286       Animation|Children's|Comedy        2286              0.356733  \n",
       "3509                      Action|Drama        3509              0.353293  "
      ]
     },
     "execution_count": 22,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "movies_df[\"Recommendation Score\"] = rec[0]\n",
    "movies_df.sort([\"Recommendation Score\"], ascending=False).head(20)"
   ]
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "Python 2",
   "language": "python",
   "name": "python2"
  },
  "language_info": {
   "codemirror_mode": {
    "name": "ipython",
    "version": 2
   },
   "file_extension": ".py",
   "mimetype": "text/x-python",
   "name": "python",
   "nbconvert_exporter": "python",
   "pygments_lexer": "ipython2",
   "version": "2.7.11"
  },
  "widgets": {
   "state": {},
   "version": "1.1.2"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 0
}