normalizing flow toy distribution name fix

Andrey Voynov · Andrey Voynov · commit 2f47b5aa5156 · 2021-03-23T15:42:10.000+03:00
diff --git a/seminar06-gen_models_1/gen_day_1.ipynb b/seminar06-gen_models_1/gen_day_1.ipynb
@@ -566,22 +566,17 @@
     }
    ],
    "source": [
-    "class Rectangle():\n",
-    "    def __init__(self, w, h, center):\n",
-    "        self.w = w\n",
-    "        self.h = h\n",
+    "class Circle():\n",
+    "    def __init__(self, center):\n",
     "        self.center = center\n",
     "\n",
     "    def __call__(self, size):\n",
     "        n = torch.randn([size, 2])\n",
     "        n = n / n.norm(dim=-1).unsqueeze(1)\n",
     "        return n + self.center + 0.1 * torch.randn([size, 2])\n",
-    "#         x = self.w * torch.rand([size, 1])\n",
-    "#         y = self.h * torch.rand([size, 1])\n",
-    "#         return torch.cat([x, y], dim=1) + self.center\n",
     "\n",
     "\n",
-    "distribution = Rectangle(1, 4, torch.tensor([4, 4]))\n",
+    "distribution = Circle(torch.tensor([4, 4]))\n",
     "ys = distribution(1024)\n",
     "\n",
     "nf = NormalizingFlow(\n",
