diff --git a/site/en/tutorials/images/transfer_learning.ipynb b/site/en/tutorials/images/transfer_learning.ipynb index bb0d9f6ea3..172bb2700b 100644 --- a/site/en/tutorials/images/transfer_learning.ipynb +++ b/site/en/tutorials/images/transfer_learning.ipynb @@ -1049,7 +1049,6 @@ "# Retrieve a batch of images from the test set\n", "image_batch, label_batch = test_dataset.as_numpy_iterator().next()\n", "predictions = model.predict_on_batch(image_batch).flatten()\n", - "# Apply a sigmoid since our model returns logits\n", "predictions = tf.where(predictions < 0.5, 0, 1)\n", "\n", "print('Predictions:\\n', predictions.numpy())\n",