Fixing the wrong predictions in transfer_learning.ipynb

tensorflow · Jun 17, 2024 · 38c16cc · 38c16cc
1 parent 75b2672
commit 38c16cc
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/site/en/tutorials/images/transfer_learning.ipynb b/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",