Merge pull request falloutdurham#44 from MarcusFra/ch05_eval

Add model.eval()

Author: falloutdurham

chapter5/Chapter 5.ipynb

@@ -7,6 +7,21 @@
     "# Chapter 5: Text Classification"
    ]
   },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "outputs": [],
+   "source": [
+    "!pip install torchtext~=0.7.0\n",
+    "!pip install torch~=1.6"
+   ],
+   "metadata": {
+    "collapsed": false,
+    "pycharm": {
+     "name": "#%%\n"
+    }
+   }
+  },
   {
    "cell_type": "code",
    "execution_count": 37,
@@ -309,6 +324,7 @@
     "    categories = {0: \"Negative\", 1:\"Positive\"}\n",
     "    processed = TWEET.process([TWEET.preprocess(tweet)])\n",
     "    processed = processed.to(device)\n",
+    "    model.eval()\n",
     "    return categories[model(processed).argmax().item()]"
    ]
   },