make predictions with the CNN

Author: vyinkabanjo

lab2/solutions/PT_Part1_MNIST_Solution.ipynb

@@ -779,7 +779,7 @@
       "source": [
         "### Make predictions with the CNN model\n",
         "\n",
-        "With the model trained, we can use it to make predictions about some images. The [`predict`](https://www.tensorflow.org/api_docs/python/tf/keras/models/Sequential#predict) function call generates the output predictions given a set of input samples.\n"
+        "With the model trained, we can use it to make predictions about some images."
       ]
     },
     {
@@ -790,7 +790,10 @@
       },
       "outputs": [],
       "source": [
-        "predictions = cnn_model.predict(test_images)"
+        "test_image, test_label = test_dataset[0]\n",
+        "test_image = test_image.unsqueeze(0)\n",
+        "cnn_model.eval()\n",
+        "predictions_test_image = cnn_model(test_image)"
       ]
     },
     {
@@ -799,7 +802,7 @@
         "id": "x9Kk1voUCaXJ"
       },
       "source": [
-        "With this function call, the model has predicted the label for each image in the testing set. Let's take a look at the prediction for the first image in the test dataset:"
+        "With this function call, the model has predicted the label of the first image in the testing set. Let's take a look at the prediction:"
       ]
     },
     {
@@ -810,7 +813,7 @@
       },
       "outputs": [],
       "source": [
-        "predictions[0]"
+        "predictions_test_image"
       ]
     },
     {
@@ -834,9 +837,9 @@
       "source": [
         "'''TODO: identify the digit with the highest confidence prediction for the first\n",
         "    image in the test dataset. '''\n",
-        "prediction = np.argmax(predictions[0])\n",
+        "predictions_value = predictions_test_image.detach().numpy()\n",
+        "prediction = np.argmax(predictions_value)\n",
         "# prediction = # TODO\n",
-        "\n",
         "print(prediction)"
       ]
     },
@@ -857,8 +860,8 @@
       },
       "outputs": [],
       "source": [
-        "print(\"Label of this digit is:\", test_labels[0])\n",
-        "plt.imshow(test_images[0,:,:,0], cmap=plt.cm.binary)\n",
+        "print(\"Label of this digit is:\", test_label)\n",
+        "plt.imshow(test_image[0,0,:,:].cpu(), cmap=plt.cm.binary)\n",
         "comet_model_2.log_figure(figure=plt)"
       ]
     },