From bf7843fe387786e07613f61a9e4c60fae2c88d02 Mon Sep 17 00:00:00 2001
From: Aashish Chaubey <chaubey.aashish@in.ibm.com>
Date: Tue, 1 Jun 2021 17:20:39 +0530
Subject: [PATCH] Changing the activation function for final layer

---
 .../cnn_cifar10_dataset.ipynb                               | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/16_cnn_cifar10_small_image_classification/cnn_cifar10_dataset.ipynb b/16_cnn_cifar10_small_image_classification/cnn_cifar10_dataset.ipynb
index 594164a..e74045f 100644
--- a/16_cnn_cifar10_small_image_classification/cnn_cifar10_dataset.ipynb
+++ b/16_cnn_cifar10_small_image_classification/cnn_cifar10_dataset.ipynb
@@ -315,7 +315,7 @@
     "        layers.Flatten(input_shape=(32,32,3)),\n",
     "        layers.Dense(3000, activation='relu'),\n",
     "        layers.Dense(1000, activation='relu'),\n",
-    "        layers.Dense(10, activation='sigmoid')    \n",
+    "        layers.Dense(10, activation='softmax')    \n",
     "    ])\n",
     "\n",
     "ann.compile(optimizer='SGD',\n",
@@ -664,9 +664,9 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.8.5"
+   "version": "3.7.3"
   }
  },
  "nbformat": 4,
  "nbformat_minor": 4
-}
\ No newline at end of file
+}