Fix for Issue phlippe#11

phlippe#11
emanbuc · Oct 19, 2021 · 4966765 · 4966765
1 parent 7cb5765
commit 4966765
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/docs/tutorial_notebooks/tutorial2/Introduction_to_PyTorch.ipynb b/docs/tutorial_notebooks/tutorial2/Introduction_to_PyTorch.ipynb
@@ -3809,9 +3809,9 @@
     "    model_inputs = torch.stack([xx1, xx2], dim=-1)\n",
     "    preds = model(model_inputs)\n",
     "    preds = torch.sigmoid(preds)\n",
-    "    output_image = preds * c0[None,None] + (1 - preds) * c1[None,None] # Specifying \"None\" in a dimension creates a new one\n",
+    "    output_image = (1 - preds) * c0[None,None] + preds * c1[None,None] # Specifying \"None\" in a dimension creates a new one\n",
     "    output_image = output_image.cpu().numpy() # Convert to numpy array. This only works for tensors on CPU, hence first push to CPU\n",
-    "    plt.imshow(output_image, origin='upper', extent=(-0.5, 1.5, -0.5, 1.5))\n",
+    "    plt.imshow(output_image, origin='lower', extent=(-0.5, 1.5, -0.5, 1.5))\n",
     "    plt.grid(False)\n",
     "\n",
     "visualize_classification(model, dataset.data, dataset.label)\n",