lab 2 edits

Author: slolla

lab2/Part2_Debiasing.ipynb

@@ -997,23 +997,24 @@
       "source": [
         "test_faces = np.asarray(test_faces)\n",
         "test_faces = test_faces.reshape(20, 64, 64, 3)\n",
-        "logits = dbvae.predict(test_faces)\n",
         "y_logit, z_mean, z_logsigma, x_recon = dbvae(test_faces)\n",
         "loss, class_loss, vae_loss = debiasing_loss_function(test_faces, x_recon, tf.ones((20, 1)), y_logit, z_mean, z_logsigma)\n",
         "indices = np.argsort(vae_loss, axis=None)\n",
         "test_faces = test_faces[indices]\n",
-        "class_loss = class_loss.numpy()[indices]\n",
+        "vae_loss = vae_loss.numpy()[indices]\n",
         "\n",
-        "\n",
-        "fig, axs = plt.subplots(1, 5)\n",
+        "fig, axs = plt.subplots(1, 5, constrained_layout=True)\n",
         "for i in range(1, 6):\n",
         "  axs[i - 1].imshow(test_faces[- 1 * i])\n",
-        "  axs[i - 1].set_title(round(vae_loss.numpy()[-1 * i], 7))\n",
+        "  axs[i - 1].set_title(round(vae_loss[-1 * i], 7))\n",
+        "print(\"Samples with the highest reconstruction loss\")\n",
+        "plt.show()\n",
         "\n",
-        "fig, axs = plt.subplots(1, 5)\n",
+        "fig, axs = plt.subplots(1, 5, constrained_layout=True)\n",
         "for i in range(5):\n",
         "  axs[i].imshow(test_faces[i])\n",
-        "  axs[i].set_title(round(vae_loss.numpy()[i], 7))"
+        "  axs[i].set_title(round(vae_loss[i], 7))\n",
+        "print(\"Samples with the lowest reconstruction loss\")"
       ]
     },
     {