Evaluate the CNN Model

Author: vyinkabanjo

lab2/solutions/PT_Part1_MNIST_Solution.ipynb

@@ -737,6 +737,13 @@
         "comet_model_2.log_figure(figure=plt)"
       ]
     },
+    {
+      "cell_type": "markdown",
+      "metadata": {},
+      "source": [
+        "### Evaluate the CNN Model"
+      ]
+    },
     {
       "cell_type": "code",
       "execution_count": null,
@@ -745,9 +752,10 @@
       },
       "outputs": [],
       "source": [
-        "'''TODO: Use the evaluate method to test the model!'''\n",
-        "test_loss, test_acc = cnn_model.evaluate(test_images, test_labels)\n",
-        "# test_loss, test_acc = # TODO\n",
+        "'''TODO: Evaluate the CNN model'''\n",
+        "\n",
+        "test_loss, test_acc = evaluate(cnn_model, trainset_loader, loss_function)\n",
+        "# test_loss, test_acc = TODO\n",
         "\n",
         "print('Test accuracy:', test_acc)"
       ]