Spam or Ham practice.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 1,
   "metadata": {
    "collapsed": true
   },
   "outputs": [],
   "source": [
    "import os\n",
    "import io\n",
    "import numpy\n",
    "from pandas import DataFrame\n",
    "from sklearn.feature_extraction.text import CountVectorizer\n",
    "from sklearn.naive_bayes import MultinomialNB"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 2,
   "metadata": {
    "collapsed": true
   },
   "outputs": [],
   "source": [
    "def readFiles(path):\n",
    "    for root, dirnames, filenames in os.walk(path):\n",
    "        for filename in filenames:\n",
    "            path = os.path.join(root, filename)\n",
    "\n",
    "            inBody = False\n",
    "            lines = []\n",
    "            f = io.open(path, 'r', encoding='latin1')\n",
    "            for line in f:\n",
    "                if inBody:\n",
    "                    lines.append(line)\n",
    "                elif line == '\\n':\n",
    "                    inBody = True\n",
    "            f.close()\n",
    "            message = '\\n'.join(lines)\n",
    "            yield path, message"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 3,
   "metadata": {
    "collapsed": true
   },
   "outputs": [],
   "source": [
    "def dataFrameFromDirectory(path, classification):\n",
    "    rows = []\n",
    "    index = []\n",
    "    for filename, message in readFiles(path):\n",
    "        rows.append({'message': message, 'class': classification})\n",
    "        index.append(filename)\n",
    "\n",
    "    return DataFrame(rows, index=index)\n",
    "\n",
    "data = DataFrame({'message': [], 'class': []})"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 4,
   "metadata": {
    "collapsed": true
   },
   "outputs": [],
   "source": [
    "data = data.append(dataFrameFromDirectory(\n",
    "    '/Users/Dhanashree/Desktop/Dhanashree/SCU Coursework/PythonDS/Pro/emails/spam', 'spam'))"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 5,
   "metadata": {
    "collapsed": true
   },
   "outputs": [],
   "source": [
    "data = data.append(dataFrameFromDirectory('/Users/Dhanashree/Desktop/Dhanashree/SCU Coursework/PythonDS/Pro/emails/ham', 'ham'))"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 6,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/html": [
       "<div>\n",
       "<style>\n",
       "    .dataframe thead tr:only-child th {\n",
       "        text-align: right;\n",
       "    }\n",
       "\n",
       "    .dataframe thead th {\n",
       "        text-align: left;\n",
       "    }\n",
       "\n",
       "    .dataframe tbody tr th {\n",
       "        vertical-align: top;\n",
       "    }\n",
       "</style>\n",
       "<table border=\"1\" class=\"dataframe\">\n",
       "  <thead>\n",
       "    <tr style=\"text-align: right;\">\n",
       "      <th></th>\n",
       "      <th>class</th>\n",
       "      <th>message</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>/Users/Dhanashree/Desktop/Dhanashree/SCU Coursework/PythonDS/Pro/emails/spam\\00001.7848dde101aa985090474a91ec93fcf0</th>\n",
       "      <td>spam</td>\n",
       "      <td>&lt;!DOCTYPE HTML PUBLIC \"-//W3C//DTD HTML 4.0 Tr...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>/Users/Dhanashree/Desktop/Dhanashree/SCU Coursework/PythonDS/Pro/emails/spam\\00002.d94f1b97e48ed3b553b3508d116e6a09</th>\n",
       "      <td>spam</td>\n",
       "      <td>1) Fight The Risk of Cancer!\\n\\nhttp://www.adc...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>/Users/Dhanashree/Desktop/Dhanashree/SCU Coursework/PythonDS/Pro/emails/spam\\00003.2ee33bc6eacdb11f38d052c44819ba6c</th>\n",
       "      <td>spam</td>\n",
       "      <td>1) Fight The Risk of Cancer!\\n\\nhttp://www.adc...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>/Users/Dhanashree/Desktop/Dhanashree/SCU Coursework/PythonDS/Pro/emails/spam\\00004.eac8de8d759b7e74154f142194282724</th>\n",
       "      <td>spam</td>\n",
       "      <td>##############################################...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>/Users/Dhanashree/Desktop/Dhanashree/SCU Coursework/PythonDS/Pro/emails/spam\\00005.57696a39d7d84318ce497886896bf90d</th>\n",
       "      <td>spam</td>\n",
       "      <td>I thought you might like these:\\n\\n1) Slim Dow...</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "                                                   class  \\\n",
       "/Users/Dhanashree/Desktop/Dhanashree/SCU Course...  spam   \n",
       "/Users/Dhanashree/Desktop/Dhanashree/SCU Course...  spam   \n",
       "/Users/Dhanashree/Desktop/Dhanashree/SCU Course...  spam   \n",
       "/Users/Dhanashree/Desktop/Dhanashree/SCU Course...  spam   \n",
       "/Users/Dhanashree/Desktop/Dhanashree/SCU Course...  spam   \n",
       "\n",
       "                                                                                              message  \n",
       "/Users/Dhanashree/Desktop/Dhanashree/SCU Course...  <!DOCTYPE HTML PUBLIC \"-//W3C//DTD HTML 4.0 Tr...  \n",
       "/Users/Dhanashree/Desktop/Dhanashree/SCU Course...  1) Fight The Risk of Cancer!\\n\\nhttp://www.adc...  \n",
       "/Users/Dhanashree/Desktop/Dhanashree/SCU Course...  1) Fight The Risk of Cancer!\\n\\nhttp://www.adc...  \n",
       "/Users/Dhanashree/Desktop/Dhanashree/SCU Course...  ##############################################...  \n",
       "/Users/Dhanashree/Desktop/Dhanashree/SCU Course...  I thought you might like these:\\n\\n1) Slim Dow...  "
      ]
     },
     "execution_count": 6,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "data.head()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 7,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/html": [
       "<div>\n",
       "<style>\n",
       "    .dataframe thead tr:only-child th {\n",
       "        text-align: right;\n",
       "    }\n",
       "\n",
       "    .dataframe thead th {\n",
       "        text-align: left;\n",
       "    }\n",
       "\n",
       "    .dataframe tbody tr th {\n",
       "        vertical-align: top;\n",
       "    }\n",
       "</style>\n",
       "<table border=\"1\" class=\"dataframe\">\n",
       "  <thead>\n",
       "    <tr style=\"text-align: right;\">\n",
       "      <th></th>\n",
       "      <th>class</th>\n",
       "      <th>message</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>/Users/Dhanashree/Desktop/Dhanashree/SCU Coursework/PythonDS/Pro/emails/spam\\00001.7848dde101aa985090474a91ec93fcf0</th>\n",
       "      <td>spam</td>\n",
       "      <td>&lt;!DOCTYPE HTML PUBLIC \"-//W3C//DTD HTML 4.0 Tr...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>/Users/Dhanashree/Desktop/Dhanashree/SCU Coursework/PythonDS/Pro/emails/spam\\00002.d94f1b97e48ed3b553b3508d116e6a09</th>\n",
       "      <td>spam</td>\n",
       "      <td>1) Fight The Risk of Cancer!\\n\\nhttp://www.adc...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>/Users/Dhanashree/Desktop/Dhanashree/SCU Coursework/PythonDS/Pro/emails/spam\\00003.2ee33bc6eacdb11f38d052c44819ba6c</th>\n",
       "      <td>spam</td>\n",
       "      <td>1) Fight The Risk of Cancer!\\n\\nhttp://www.adc...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>/Users/Dhanashree/Desktop/Dhanashree/SCU Coursework/PythonDS/Pro/emails/spam\\00004.eac8de8d759b7e74154f142194282724</th>\n",
       "      <td>spam</td>\n",
       "      <td>##############################################...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>/Users/Dhanashree/Desktop/Dhanashree/SCU Coursework/PythonDS/Pro/emails/spam\\00005.57696a39d7d84318ce497886896bf90d</th>\n",
       "      <td>spam</td>\n",
       "      <td>I thought you might like these:\\n\\n1) Slim Dow...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>/Users/Dhanashree/Desktop/Dhanashree/SCU Coursework/PythonDS/Pro/emails/spam\\00006.5ab5620d3d7c6c0db76234556a16f6c1</th>\n",
       "      <td>spam</td>\n",
       "      <td>A POWERHOUSE GIFTING PROGRAM You Don't Want To...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>/Users/Dhanashree/Desktop/Dhanashree/SCU Coursework/PythonDS/Pro/emails/spam\\00007.d8521faf753ff9ee989122f6816f87d7</th>\n",
       "      <td>spam</td>\n",
       "      <td>Help wanted.  We are a 14 year old fortune 500...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>/Users/Dhanashree/Desktop/Dhanashree/SCU Coursework/PythonDS/Pro/emails/spam\\00008.dfd941deb10f5eed78b1594b131c9266</th>\n",
       "      <td>spam</td>\n",
       "      <td>&lt;html&gt;\\n\\n&lt;head&gt;\\n\\n&lt;title&gt;ReliaQuote - Save U...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>/Users/Dhanashree/Desktop/Dhanashree/SCU Coursework/PythonDS/Pro/emails/spam\\00009.027bf6e0b0c4ab34db3ce0ea4bf2edab</th>\n",
       "      <td>spam</td>\n",
       "      <td>TIRED OF THE BULL OUT THERE?\\n\\nWant To Stop L...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>/Users/Dhanashree/Desktop/Dhanashree/SCU Coursework/PythonDS/Pro/emails/spam\\00010.445affef4c70feec58f9198cfbc22997</th>\n",
       "      <td>spam</td>\n",
       "      <td>Dear ricardo1 ,\\n\\n\\n\\n&lt;html&gt;\\n\\n&lt;body&gt;\\n\\n&lt;ce...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>/Users/Dhanashree/Desktop/Dhanashree/SCU Coursework/PythonDS/Pro/emails/spam\\00011.61816b9ad167657773a427d890d0468e</th>\n",
       "      <td>spam</td>\n",
       "      <td>Cellular Phone Accessories All At Below Wholes...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>/Users/Dhanashree/Desktop/Dhanashree/SCU Coursework/PythonDS/Pro/emails/spam\\00012.381e4f512915109ba1e0853a7a8407b2</th>\n",
       "      <td>spam</td>\n",
       "      <td>&lt;table width=\"600\" border=\"20\" align=\"center\" ...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>/Users/Dhanashree/Desktop/Dhanashree/SCU Coursework/PythonDS/Pro/emails/spam\\00013.d3f0b591a65f116ea5d9d4ad919f83aa</th>\n",
       "      <td>spam</td>\n",
       "      <td>1) Fight The Risk of Cancer!\\n\\nhttp://www.adc...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>/Users/Dhanashree/Desktop/Dhanashree/SCU Coursework/PythonDS/Pro/emails/spam\\00014.7d38c46424f24fc8012ac15a95a2ac14</th>\n",
       "      <td>spam</td>\n",
       "      <td>&lt;HTML&gt;&lt;HEAD&gt;&lt;TITLE&gt;FREE Motorola Cell Phone wi...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>/Users/Dhanashree/Desktop/Dhanashree/SCU Coursework/PythonDS/Pro/emails/spam\\00015.048434ab64c86cf890eda1326a5643f5</th>\n",
       "      <td>spam</td>\n",
       "      <td>&lt;HTML&gt;&lt;HEAD&gt;&lt;TITLE&gt;Lowest Rate Services&lt;/TITLE...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>/Users/Dhanashree/Desktop/Dhanashree/SCU Coursework/PythonDS/Pro/emails/spam\\00016.67fb281761ca1051a22ec3f21917e7c0</th>\n",
       "      <td>spam</td>\n",
       "      <td>\\n\\n\\n\\nWant to watch Sporting Events?--Movies...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>/Users/Dhanashree/Desktop/Dhanashree/SCU Coursework/PythonDS/Pro/emails/spam\\00017.1a938ecddd047b93cbd7ed92c241e6d1</th>\n",
       "      <td>spam</td>\n",
       "      <td>Help wanted.  We are a 14 year old fortune 500...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>/Users/Dhanashree/Desktop/Dhanashree/SCU Coursework/PythonDS/Pro/emails/spam\\00018.5b2765c42b7648d41c93b9b27140b23a</th>\n",
       "      <td>spam</td>\n",
       "      <td>DEAR FRIEND,I AM MRS.  SESE-SEKO WIDOW OF LATE...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>/Users/Dhanashree/Desktop/Dhanashree/SCU Coursework/PythonDS/Pro/emails/spam\\00019.bbc97ad616ffd06e93ce0f821ca8c381</th>\n",
       "      <td>spam</td>\n",
       "      <td>Lowest rates available for term life insurance...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>/Users/Dhanashree/Desktop/Dhanashree/SCU Coursework/PythonDS/Pro/emails/spam\\00020.29725cf331fc21e18a1809e7d8b27332</th>\n",
       "      <td>spam</td>\n",
       "      <td>1) Fight The Risk of Cancer!\\n\\nhttp://www.adc...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>/Users/Dhanashree/Desktop/Dhanashree/SCU Coursework/PythonDS/Pro/emails/spam\\00021.effe1449462a9d7ad7af0f1c94b1a237</th>\n",
       "      <td>spam</td>\n",
       "      <td>CENTRAL BANK OF NIGERIA\\n\\nFOREIGN REMITTANCE ...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>/Users/Dhanashree/Desktop/Dhanashree/SCU Coursework/PythonDS/Pro/emails/spam\\00022.8203cdf03888f656dc0381701148f73d</th>\n",
       "      <td>spam</td>\n",
       "      <td>--===_SecAtt_000_1fheucnqggtggp\\n\\nContent-Typ...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>/Users/Dhanashree/Desktop/Dhanashree/SCU Coursework/PythonDS/Pro/emails/spam\\00023.b6d27c684f5fc803cfa1060adb2d0805</th>\n",
       "      <td>spam</td>\n",
       "      <td>------=_NextPart_000_00B2_83B03D1E.C6530E24\\n\\...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>/Users/Dhanashree/Desktop/Dhanashree/SCU Coursework/PythonDS/Pro/emails/spam\\00024.6b5437b14d403176c3f046c871b5b52f</th>\n",
       "      <td>spam</td>\n",
       "      <td>This is a multi-part message in MIME format.\\n...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>/Users/Dhanashree/Desktop/Dhanashree/SCU Coursework/PythonDS/Pro/emails/spam\\00025.619ab8051359048795e3cd09e82ad1a0</th>\n",
       "      <td>spam</td>\n",
       "      <td>&lt;HTML&gt;&lt;HEAD&gt;\\n\\n&lt;META http-equiv=3DContent-Typ...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>/Users/Dhanashree/Desktop/Dhanashree/SCU Coursework/PythonDS/Pro/emails/spam\\00026.da18dbed27ae933172f7a70f860c6ad0</th>\n",
       "      <td>spam</td>\n",
       "      <td>DEAR FRIEND,I AM MRS.  SESE-SEKO WIDOW OF LATE...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>/Users/Dhanashree/Desktop/Dhanashree/SCU Coursework/PythonDS/Pro/emails/spam\\00027.d1d0f97e096fe08fc80a4939355759e7</th>\n",
       "      <td>spam</td>\n",
       "      <td>1) Fight The Risk of Cancer!\\n\\nhttp://www.adc...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>/Users/Dhanashree/Desktop/Dhanashree/SCU Coursework/PythonDS/Pro/emails/spam\\00028.ace98eff213f4e6314b5571aece625e1</th>\n",
       "      <td>spam</td>\n",
       "      <td>&lt;HTML&gt;&lt;HEAD&gt;&lt;TITLE&gt;MILFhunter&lt;/TITLE&gt;\\n\\n&lt;META...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>/Users/Dhanashree/Desktop/Dhanashree/SCU Coursework/PythonDS/Pro/emails/spam\\00029.de865ad8d5ad0df985ae2f72388befba</th>\n",
       "      <td>spam</td>\n",
       "      <td>&lt;html&gt;\\n\\n&lt;head&gt;\\n\\n&lt;/head&gt;\\n\\n&lt;body&gt;\\n\\n\\n\\n&lt;...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>/Users/Dhanashree/Desktop/Dhanashree/SCU Coursework/PythonDS/Pro/emails/spam\\00030.0c9cdd9d4025bd55dac02719ec8d29dc</th>\n",
       "      <td>spam</td>\n",
       "      <td>&lt;html&gt;\\n\\n\\n\\n&lt;head&gt;\\n\\n&lt;meta http-equiv=3D\"Co...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>...</th>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>/Users/Dhanashree/Desktop/Dhanashree/SCU Coursework/PythonDS/Pro/emails/ham\\02471.18281d43dc0775e915267c2ea5170f1f</th>\n",
       "      <td>ham</td>\n",
       "      <td>This is possible, however using SA as a block ...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>/Users/Dhanashree/Desktop/Dhanashree/SCU Coursework/PythonDS/Pro/emails/ham\\02472.5c879dd55c3d4171e1787e8529bbd7e1</th>\n",
       "      <td>ham</td>\n",
       "      <td>\\n\\n--- Martin Adamson &lt;martin@srv0.ems.ed.ac....</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>/Users/Dhanashree/Desktop/Dhanashree/SCU Coursework/PythonDS/Pro/emails/ham\\02473.207afa13ad7d745dfd1344f84531ac16</th>\n",
       "      <td>ham</td>\n",
       "      <td>----- Original Message -----\\n\\nFrom: \"Tim Cha...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>/Users/Dhanashree/Desktop/Dhanashree/SCU Coursework/PythonDS/Pro/emails/ham\\02474.c76ffef81a2529389e6c3bbb172184d7</th>\n",
       "      <td>ham</td>\n",
       "      <td>\\n\\n&gt; Mr Tim Chapman, freelance gentleman of l...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>/Users/Dhanashree/Desktop/Dhanashree/SCU Coursework/PythonDS/Pro/emails/ham\\02475.9277ee243e3f51fa53ed6be55798d360</th>\n",
       "      <td>ham</td>\n",
       "      <td>Smith, Graham - Computing Technician wrote:\\n\\...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>/Users/Dhanashree/Desktop/Dhanashree/SCU Coursework/PythonDS/Pro/emails/ham\\02476.de1d459426662492dd1235046b504c3d</th>\n",
       "      <td>ham</td>\n",
       "      <td>Geege wrote a strange story:\\n\\n&gt;I know a guy ...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>/Users/Dhanashree/Desktop/Dhanashree/SCU Coursework/PythonDS/Pro/emails/ham\\02477.07b2069e9827cfd6f97d07eea2913d57</th>\n",
       "      <td>ham</td>\n",
       "      <td>\\n\\n[Paul Moore]\\n\\n&gt;    but let's walk before...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>/Users/Dhanashree/Desktop/Dhanashree/SCU Coursework/PythonDS/Pro/emails/ham\\02478.40723f38488bddaf5a24ef2a91679c75</th>\n",
       "      <td>ham</td>\n",
       "      <td>On Mon, Nov 25, 2002 at 06:54:49PM +0000, Phil...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>/Users/Dhanashree/Desktop/Dhanashree/SCU Coursework/PythonDS/Pro/emails/ham\\02479.14365bcad3a60fcf24c5c1813f6291fb</th>\n",
       "      <td>ham</td>\n",
       "      <td>\\n\\nI don't know how one can expect better and...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>/Users/Dhanashree/Desktop/Dhanashree/SCU Coursework/PythonDS/Pro/emails/ham\\02480.72714df60c9be29d6f7985c777cbfc13</th>\n",
       "      <td>ham</td>\n",
       "      <td>No, you need to learn how declarations work in...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>/Users/Dhanashree/Desktop/Dhanashree/SCU Coursework/PythonDS/Pro/emails/ham\\02481.176b368fe4b90682f33647d65a8b97a3</th>\n",
       "      <td>ham</td>\n",
       "      <td>\\n\\n    Richie&gt; As I understand it, post-1.8x ...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>/Users/Dhanashree/Desktop/Dhanashree/SCU Coursework/PythonDS/Pro/emails/ham\\02482.35c166ec6a85e108ad693ea43329762f</th>\n",
       "      <td>ham</td>\n",
       "      <td>\\n\\n    Paul&gt; I suspect the best answer is to ...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>/Users/Dhanashree/Desktop/Dhanashree/SCU Coursework/PythonDS/Pro/emails/ham\\02483.ab1bee02c10ddecc0e86c39eaebc2996</th>\n",
       "      <td>ham</td>\n",
       "      <td>The Times\\n\\n\\n\\n \\n\\n December 04, 2002 \\n\\n ...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>/Users/Dhanashree/Desktop/Dhanashree/SCU Coursework/PythonDS/Pro/emails/ham\\02484.32a0bca2600788be144b93cae341efbf</th>\n",
       "      <td>ham</td>\n",
       "      <td>I have to say I was surprised about Jacko dang...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>/Users/Dhanashree/Desktop/Dhanashree/SCU Coursework/PythonDS/Pro/emails/ham\\02485.ba9aebbdbec0d9fecec595eeebe5db87</th>\n",
       "      <td>ham</td>\n",
       "      <td>Now then I recently read a novel about exactly...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>/Users/Dhanashree/Desktop/Dhanashree/SCU Coursework/PythonDS/Pro/emails/ham\\02486.bdf90e871b673fd14f47f3fe36622742</th>\n",
       "      <td>ham</td>\n",
       "      <td>What the hell is it with these mini remote con...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>/Users/Dhanashree/Desktop/Dhanashree/SCU Coursework/PythonDS/Pro/emails/ham\\02487.c2e725d509201dc30debb7bd94d07f5e</th>\n",
       "      <td>ham</td>\n",
       "      <td>here, for your enjoyment, is a little somethin...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>/Users/Dhanashree/Desktop/Dhanashree/SCU Coursework/PythonDS/Pro/emails/ham\\02488.68fed64ff8169f1505b74080bb7b6158</th>\n",
       "      <td>ham</td>\n",
       "      <td>Sean O'Donnell wrote:\\n\\n&gt; Doesnt answer your ...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>/Users/Dhanashree/Desktop/Dhanashree/SCU Coursework/PythonDS/Pro/emails/ham\\02489.85c20a6f9d75714d9f44398baeddd416</th>\n",
       "      <td>ham</td>\n",
       "      <td>Joe McNally writes:\\n\\n\\n\\n&gt; What the hell is ...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>/Users/Dhanashree/Desktop/Dhanashree/SCU Coursework/PythonDS/Pro/emails/ham\\02490.7be0f683db6994ddd8445cdcc2eb5042</th>\n",
       "      <td>ham</td>\n",
       "      <td>http://news.bbc.co.uk/1/hi/world/europe/254182...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>/Users/Dhanashree/Desktop/Dhanashree/SCU Coursework/PythonDS/Pro/emails/ham\\02491.c26245be2a5096fa86647d594561c511</th>\n",
       "      <td>ham</td>\n",
       "      <td>Hi all.\\n\\nDoes anyone know how to set up dual...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>/Users/Dhanashree/Desktop/Dhanashree/SCU Coursework/PythonDS/Pro/emails/ham\\02492.6aede44f654a1bbc60c95c7dd770e624</th>\n",
       "      <td>ham</td>\n",
       "      <td>Carlos Luna wrote:\\n\\n\\n\\n&gt;Hi all.\\n\\n&gt;Does an...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>/Users/Dhanashree/Desktop/Dhanashree/SCU Coursework/PythonDS/Pro/emails/ham\\02493.f9f2870094430b7db8b0c1052b302cf1</th>\n",
       "      <td>ham</td>\n",
       "      <td>Hi all\\n\\n\\n\\n\\n\\nI have a prob when trying to...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>/Users/Dhanashree/Desktop/Dhanashree/SCU Coursework/PythonDS/Pro/emails/ham\\02494.a14f2d3a9bef3f59aa419b03aee8f871</th>\n",
       "      <td>ham</td>\n",
       "      <td>Tim Chapman writes:\\n\\n\\n\\n&gt; http://news.bbc.c...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>/Users/Dhanashree/Desktop/Dhanashree/SCU Coursework/PythonDS/Pro/emails/ham\\02495.5064946e77b3046873da91fc47656465</th>\n",
       "      <td>ham</td>\n",
       "      <td>&gt; I had the same problem when installing Win o...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>/Users/Dhanashree/Desktop/Dhanashree/SCU Coursework/PythonDS/Pro/emails/ham\\02496.aae0c81581895acfe65323f344340856</th>\n",
       "      <td>ham</td>\n",
       "      <td>Man killed 'trying to surf' on Tube train \\n\\n...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>/Users/Dhanashree/Desktop/Dhanashree/SCU Coursework/PythonDS/Pro/emails/ham\\02497.60497db0a06c2132ec2374b2898084d3</th>\n",
       "      <td>ham</td>\n",
       "      <td>Hi Gianni,\\n\\n\\n\\nA very good resource for thi...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>/Users/Dhanashree/Desktop/Dhanashree/SCU Coursework/PythonDS/Pro/emails/ham\\02498.09835f512f156da210efb99fcc523e21</th>\n",
       "      <td>ham</td>\n",
       "      <td>Gianni Ponzi wrote:\\n\\n&gt; I have a prob when tr...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>/Users/Dhanashree/Desktop/Dhanashree/SCU Coursework/PythonDS/Pro/emails/ham\\02499.b4af165650f138b10f9941f6cc5bce3c</th>\n",
       "      <td>ham</td>\n",
       "      <td>Neale Pickett &lt;neale@woozle.org&gt; writes:\\n\\n\\n...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>/Users/Dhanashree/Desktop/Dhanashree/SCU Coursework/PythonDS/Pro/emails/ham\\02500.05b3496ce7bca306bed0805425ec8621</th>\n",
       "      <td>ham</td>\n",
       "      <td>\\n\\nHi,\\n\\n\\n\\nI think you need to give us a l...</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "<p>3000 rows × 2 columns</p>\n",
       "</div>"
      ],
      "text/plain": [
       "                                                   class  \\\n",
       "/Users/Dhanashree/Desktop/Dhanashree/SCU Course...  spam   \n",
       "/Users/Dhanashree/Desktop/Dhanashree/SCU Course...  spam   \n",
       "/Users/Dhanashree/Desktop/Dhanashree/SCU Course...  spam   \n",
       "/Users/Dhanashree/Desktop/Dhanashree/SCU Course...  spam   \n",
       "/Users/Dhanashree/Desktop/Dhanashree/SCU Course...  spam   \n",
       "/Users/Dhanashree/Desktop/Dhanashree/SCU Course...  spam   \n",
       "/Users/Dhanashree/Desktop/Dhanashree/SCU Course...  spam   \n",
       "/Users/Dhanashree/Desktop/Dhanashree/SCU Course...  spam   \n",
       "/Users/Dhanashree/Desktop/Dhanashree/SCU Course...  spam   \n",
       "/Users/Dhanashree/Desktop/Dhanashree/SCU Course...  spam   \n",
       "/Users/Dhanashree/Desktop/Dhanashree/SCU Course...  spam   \n",
       "/Users/Dhanashree/Desktop/Dhanashree/SCU Course...  spam   \n",
       "/Users/Dhanashree/Desktop/Dhanashree/SCU Course...  spam   \n",
       "/Users/Dhanashree/Desktop/Dhanashree/SCU Course...  spam   \n",
       "/Users/Dhanashree/Desktop/Dhanashree/SCU Course...  spam   \n",
       "/Users/Dhanashree/Desktop/Dhanashree/SCU Course...  spam   \n",
       "/Users/Dhanashree/Desktop/Dhanashree/SCU Course...  spam   \n",
       "/Users/Dhanashree/Desktop/Dhanashree/SCU Course...  spam   \n",
       "/Users/Dhanashree/Desktop/Dhanashree/SCU Course...  spam   \n",
       "/Users/Dhanashree/Desktop/Dhanashree/SCU Course...  spam   \n",
       "/Users/Dhanashree/Desktop/Dhanashree/SCU Course...  spam   \n",
       "/Users/Dhanashree/Desktop/Dhanashree/SCU Course...  spam   \n",
       "/Users/Dhanashree/Desktop/Dhanashree/SCU Course...  spam   \n",
       "/Users/Dhanashree/Desktop/Dhanashree/SCU Course...  spam   \n",
       "/Users/Dhanashree/Desktop/Dhanashree/SCU Course...  spam   \n",
       "/Users/Dhanashree/Desktop/Dhanashree/SCU Course...  spam   \n",
       "/Users/Dhanashree/Desktop/Dhanashree/SCU Course...  spam   \n",
       "/Users/Dhanashree/Desktop/Dhanashree/SCU Course...  spam   \n",
       "/Users/Dhanashree/Desktop/Dhanashree/SCU Course...  spam   \n",
       "/Users/Dhanashree/Desktop/Dhanashree/SCU Course...  spam   \n",
       "...                                                  ...   \n",
       "/Users/Dhanashree/Desktop/Dhanashree/SCU Course...   ham   \n",
       "/Users/Dhanashree/Desktop/Dhanashree/SCU Course...   ham   \n",
       "/Users/Dhanashree/Desktop/Dhanashree/SCU Course...   ham   \n",
       "/Users/Dhanashree/Desktop/Dhanashree/SCU Course...   ham   \n",
       "/Users/Dhanashree/Desktop/Dhanashree/SCU Course...   ham   \n",
       "/Users/Dhanashree/Desktop/Dhanashree/SCU Course...   ham   \n",
       "/Users/Dhanashree/Desktop/Dhanashree/SCU Course...   ham   \n",
       "/Users/Dhanashree/Desktop/Dhanashree/SCU Course...   ham   \n",
       "/Users/Dhanashree/Desktop/Dhanashree/SCU Course...   ham   \n",
       "/Users/Dhanashree/Desktop/Dhanashree/SCU Course...   ham   \n",
       "/Users/Dhanashree/Desktop/Dhanashree/SCU Course...   ham   \n",
       "/Users/Dhanashree/Desktop/Dhanashree/SCU Course...   ham   \n",
       "/Users/Dhanashree/Desktop/Dhanashree/SCU Course...   ham   \n",
       "/Users/Dhanashree/Desktop/Dhanashree/SCU Course...   ham   \n",
       "/Users/Dhanashree/Desktop/Dhanashree/SCU Course...   ham   \n",
       "/Users/Dhanashree/Desktop/Dhanashree/SCU Course...   ham   \n",
       "/Users/Dhanashree/Desktop/Dhanashree/SCU Course...   ham   \n",
       "/Users/Dhanashree/Desktop/Dhanashree/SCU Course...   ham   \n",
       "/Users/Dhanashree/Desktop/Dhanashree/SCU Course...   ham   \n",
       "/Users/Dhanashree/Desktop/Dhanashree/SCU Course...   ham   \n",
       "/Users/Dhanashree/Desktop/Dhanashree/SCU Course...   ham   \n",
       "/Users/Dhanashree/Desktop/Dhanashree/SCU Course...   ham   \n",
       "/Users/Dhanashree/Desktop/Dhanashree/SCU Course...   ham   \n",
       "/Users/Dhanashree/Desktop/Dhanashree/SCU Course...   ham   \n",
       "/Users/Dhanashree/Desktop/Dhanashree/SCU Course...   ham   \n",
       "/Users/Dhanashree/Desktop/Dhanashree/SCU Course...   ham   \n",
       "/Users/Dhanashree/Desktop/Dhanashree/SCU Course...   ham   \n",
       "/Users/Dhanashree/Desktop/Dhanashree/SCU Course...   ham   \n",
       "/Users/Dhanashree/Desktop/Dhanashree/SCU Course...   ham   \n",
       "/Users/Dhanashree/Desktop/Dhanashree/SCU Course...   ham   \n",
       "\n",
       "                                                                                              message  \n",
       "/Users/Dhanashree/Desktop/Dhanashree/SCU Course...  <!DOCTYPE HTML PUBLIC \"-//W3C//DTD HTML 4.0 Tr...  \n",
       "/Users/Dhanashree/Desktop/Dhanashree/SCU Course...  1) Fight The Risk of Cancer!\\n\\nhttp://www.adc...  \n",
       "/Users/Dhanashree/Desktop/Dhanashree/SCU Course...  1) Fight The Risk of Cancer!\\n\\nhttp://www.adc...  \n",
       "/Users/Dhanashree/Desktop/Dhanashree/SCU Course...  ##############################################...  \n",
       "/Users/Dhanashree/Desktop/Dhanashree/SCU Course...  I thought you might like these:\\n\\n1) Slim Dow...  \n",
       "/Users/Dhanashree/Desktop/Dhanashree/SCU Course...  A POWERHOUSE GIFTING PROGRAM You Don't Want To...  \n",
       "/Users/Dhanashree/Desktop/Dhanashree/SCU Course...  Help wanted.  We are a 14 year old fortune 500...  \n",
       "/Users/Dhanashree/Desktop/Dhanashree/SCU Course...  <html>\\n\\n<head>\\n\\n<title>ReliaQuote - Save U...  \n",
       "/Users/Dhanashree/Desktop/Dhanashree/SCU Course...  TIRED OF THE BULL OUT THERE?\\n\\nWant To Stop L...  \n",
       "/Users/Dhanashree/Desktop/Dhanashree/SCU Course...  Dear ricardo1 ,\\n\\n\\n\\n<html>\\n\\n<body>\\n\\n<ce...  \n",
       "/Users/Dhanashree/Desktop/Dhanashree/SCU Course...  Cellular Phone Accessories All At Below Wholes...  \n",
       "/Users/Dhanashree/Desktop/Dhanashree/SCU Course...  <table width=\"600\" border=\"20\" align=\"center\" ...  \n",
       "/Users/Dhanashree/Desktop/Dhanashree/SCU Course...  1) Fight The Risk of Cancer!\\n\\nhttp://www.adc...  \n",
       "/Users/Dhanashree/Desktop/Dhanashree/SCU Course...  <HTML><HEAD><TITLE>FREE Motorola Cell Phone wi...  \n",
       "/Users/Dhanashree/Desktop/Dhanashree/SCU Course...  <HTML><HEAD><TITLE>Lowest Rate Services</TITLE...  \n",
       "/Users/Dhanashree/Desktop/Dhanashree/SCU Course...  \\n\\n\\n\\nWant to watch Sporting Events?--Movies...  \n",
       "/Users/Dhanashree/Desktop/Dhanashree/SCU Course...  Help wanted.  We are a 14 year old fortune 500...  \n",
       "/Users/Dhanashree/Desktop/Dhanashree/SCU Course...  DEAR FRIEND,I AM MRS.  SESE-SEKO WIDOW OF LATE...  \n",
       "/Users/Dhanashree/Desktop/Dhanashree/SCU Course...  Lowest rates available for term life insurance...  \n",
       "/Users/Dhanashree/Desktop/Dhanashree/SCU Course...  1) Fight The Risk of Cancer!\\n\\nhttp://www.adc...  \n",
       "/Users/Dhanashree/Desktop/Dhanashree/SCU Course...  CENTRAL BANK OF NIGERIA\\n\\nFOREIGN REMITTANCE ...  \n",
       "/Users/Dhanashree/Desktop/Dhanashree/SCU Course...  --===_SecAtt_000_1fheucnqggtggp\\n\\nContent-Typ...  \n",
       "/Users/Dhanashree/Desktop/Dhanashree/SCU Course...  ------=_NextPart_000_00B2_83B03D1E.C6530E24\\n\\...  \n",
       "/Users/Dhanashree/Desktop/Dhanashree/SCU Course...  This is a multi-part message in MIME format.\\n...  \n",
       "/Users/Dhanashree/Desktop/Dhanashree/SCU Course...  <HTML><HEAD>\\n\\n<META http-equiv=3DContent-Typ...  \n",
       "/Users/Dhanashree/Desktop/Dhanashree/SCU Course...  DEAR FRIEND,I AM MRS.  SESE-SEKO WIDOW OF LATE...  \n",
       "/Users/Dhanashree/Desktop/Dhanashree/SCU Course...  1) Fight The Risk of Cancer!\\n\\nhttp://www.adc...  \n",
       "/Users/Dhanashree/Desktop/Dhanashree/SCU Course...  <HTML><HEAD><TITLE>MILFhunter</TITLE>\\n\\n<META...  \n",
       "/Users/Dhanashree/Desktop/Dhanashree/SCU Course...  <html>\\n\\n<head>\\n\\n</head>\\n\\n<body>\\n\\n\\n\\n<...  \n",
       "/Users/Dhanashree/Desktop/Dhanashree/SCU Course...  <html>\\n\\n\\n\\n<head>\\n\\n<meta http-equiv=3D\"Co...  \n",
       "...                                                                                               ...  \n",
       "/Users/Dhanashree/Desktop/Dhanashree/SCU Course...  This is possible, however using SA as a block ...  \n",
       "/Users/Dhanashree/Desktop/Dhanashree/SCU Course...  \\n\\n--- Martin Adamson <martin@srv0.ems.ed.ac....  \n",
       "/Users/Dhanashree/Desktop/Dhanashree/SCU Course...  ----- Original Message -----\\n\\nFrom: \"Tim Cha...  \n",
       "/Users/Dhanashree/Desktop/Dhanashree/SCU Course...  \\n\\n> Mr Tim Chapman, freelance gentleman of l...  \n",
       "/Users/Dhanashree/Desktop/Dhanashree/SCU Course...  Smith, Graham - Computing Technician wrote:\\n\\...  \n",
       "/Users/Dhanashree/Desktop/Dhanashree/SCU Course...  Geege wrote a strange story:\\n\\n>I know a guy ...  \n",
       "/Users/Dhanashree/Desktop/Dhanashree/SCU Course...  \\n\\n[Paul Moore]\\n\\n>    but let's walk before...  \n",
       "/Users/Dhanashree/Desktop/Dhanashree/SCU Course...  On Mon, Nov 25, 2002 at 06:54:49PM +0000, Phil...  \n",
       "/Users/Dhanashree/Desktop/Dhanashree/SCU Course...  \\n\\nI don't know how one can expect better and...  \n",
       "/Users/Dhanashree/Desktop/Dhanashree/SCU Course...  No, you need to learn how declarations work in...  \n",
       "/Users/Dhanashree/Desktop/Dhanashree/SCU Course...  \\n\\n    Richie> As I understand it, post-1.8x ...  \n",
       "/Users/Dhanashree/Desktop/Dhanashree/SCU Course...  \\n\\n    Paul> I suspect the best answer is to ...  \n",
       "/Users/Dhanashree/Desktop/Dhanashree/SCU Course...  The Times\\n\\n\\n\\n \\n\\n December 04, 2002 \\n\\n ...  \n",
       "/Users/Dhanashree/Desktop/Dhanashree/SCU Course...  I have to say I was surprised about Jacko dang...  \n",
       "/Users/Dhanashree/Desktop/Dhanashree/SCU Course...  Now then I recently read a novel about exactly...  \n",
       "/Users/Dhanashree/Desktop/Dhanashree/SCU Course...  What the hell is it with these mini remote con...  \n",
       "/Users/Dhanashree/Desktop/Dhanashree/SCU Course...  here, for your enjoyment, is a little somethin...  \n",
       "/Users/Dhanashree/Desktop/Dhanashree/SCU Course...  Sean O'Donnell wrote:\\n\\n> Doesnt answer your ...  \n",
       "/Users/Dhanashree/Desktop/Dhanashree/SCU Course...  Joe McNally writes:\\n\\n\\n\\n> What the hell is ...  \n",
       "/Users/Dhanashree/Desktop/Dhanashree/SCU Course...  http://news.bbc.co.uk/1/hi/world/europe/254182...  \n",
       "/Users/Dhanashree/Desktop/Dhanashree/SCU Course...  Hi all.\\n\\nDoes anyone know how to set up dual...  \n",
       "/Users/Dhanashree/Desktop/Dhanashree/SCU Course...  Carlos Luna wrote:\\n\\n\\n\\n>Hi all.\\n\\n>Does an...  \n",
       "/Users/Dhanashree/Desktop/Dhanashree/SCU Course...  Hi all\\n\\n\\n\\n\\n\\nI have a prob when trying to...  \n",
       "/Users/Dhanashree/Desktop/Dhanashree/SCU Course...  Tim Chapman writes:\\n\\n\\n\\n> http://news.bbc.c...  \n",
       "/Users/Dhanashree/Desktop/Dhanashree/SCU Course...  > I had the same problem when installing Win o...  \n",
       "/Users/Dhanashree/Desktop/Dhanashree/SCU Course...  Man killed 'trying to surf' on Tube train \\n\\n...  \n",
       "/Users/Dhanashree/Desktop/Dhanashree/SCU Course...  Hi Gianni,\\n\\n\\n\\nA very good resource for thi...  \n",
       "/Users/Dhanashree/Desktop/Dhanashree/SCU Course...  Gianni Ponzi wrote:\\n\\n> I have a prob when tr...  \n",
       "/Users/Dhanashree/Desktop/Dhanashree/SCU Course...  Neale Pickett <neale@woozle.org> writes:\\n\\n\\n...  \n",
       "/Users/Dhanashree/Desktop/Dhanashree/SCU Course...  \\n\\nHi,\\n\\n\\n\\nI think you need to give us a l...  \n",
       "\n",
       "[3000 rows x 2 columns]"
      ]
     },
     "execution_count": 7,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "data"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 8,
   "metadata": {
    "collapsed": true
   },
   "outputs": [],
   "source": [
    "vectorizer= CountVectorizer()\n",
    "counts= vectorizer.fit_transform(data['message'].values)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 9,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "<3000x62964 sparse matrix of type '<class 'numpy.int64'>'\n",
       "\twith 429785 stored elements in Compressed Sparse Row format>"
      ]
     },
     "execution_count": 9,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "counts"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 10,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "MultinomialNB(alpha=1.0, class_prior=None, fit_prior=True)"
      ]
     },
     "execution_count": 10,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "ClassifierModel = MultinomialNB()\n",
    "\n",
    "targets= data['class'].values\n",
    "\n",
    "ClassifierModel.fit(counts, targets)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 52,
   "metadata": {
    "collapsed": true
   },
   "outputs": [],
   "source": [
    "examples= ['Free Free Free', 'Free FREE FREE Sex', 'power of attorney request', 'meet you at 4.30 pm', \n",
    "           'Central Bank of Nigeria has 100 Million dollars for you', 'Get City card, get 71 litres of free fuel**', 'HI DHANASHREEMANE03, One Cover for every Family Member',\n",
    "          'Hi dhanashreemane03 Full-body checkup for just Rs. 999']\n",
    "\n",
    "example_counts= vectorizer.transform(examples)\n",
    "\n",
    "predictions= ClassifierModel.predict(example_counts)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 53,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "array(['spam', 'spam', 'ham', 'ham', 'spam', 'spam', 'ham', 'ham'],\n",
       "      dtype='<U4')"
      ]
     },
     "execution_count": 53,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "predictions"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {
    "collapsed": true
   },
   "outputs": [],
   "source": []
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "Python 3",
   "language": "python",
   "name": "python3"
  },
  "language_info": {
   "codemirror_mode": {
    "name": "ipython",
    "version": 3
   },
   "file_extension": ".py",
   "mimetype": "text/x-python",
   "name": "python",
   "nbconvert_exporter": "python",
   "pygments_lexer": "ipython3",
   "version": "3.6.3"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 2
}