Fixing the wrong predictions in transfer_learning.ipynb

Author: tilakrayal

site/en/tutorials/images/transfer_learning.ipynb

@@ -1051,7 +1051,6 @@
         "predictions = model.predict_on_batch(image_batch).flatten()\n",
         "\n",
         "# Apply a sigmoid since our model returns logits\n",
-        "predictions = tf.nn.sigmoid(predictions)\n",
         "predictions = tf.where(predictions < 0.5, 0, 1)\n",
         "\n",
         "print('Predictions:\\n', predictions.numpy())\n",