Skip to content

Commit 940d94c

Browse files
Merge pull request #2314 from tilakrayal:patch-16
PiperOrigin-RevId: 646695243
2 parents 75b2672 + f04e305 commit 940d94c

File tree

1 file changed

+0
-3
lines changed

1 file changed

+0
-3
lines changed

site/en/tutorials/images/transfer_learning.ipynb

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1049,9 +1049,6 @@
10491049
"# Retrieve a batch of images from the test set\n",
10501050
"image_batch, label_batch = test_dataset.as_numpy_iterator().next()\n",
10511051
"predictions = model.predict_on_batch(image_batch).flatten()\n",
1052-
"\n",
1053-
"# Apply a sigmoid since our model returns logits\n",
1054-
"predictions = tf.nn.sigmoid(predictions)\n",
10551052
"predictions = tf.where(predictions < 0.5, 0, 1)\n",
10561053
"\n",
10571054
"print('Predictions:\\n', predictions.numpy())\n",

0 commit comments

Comments
 (0)