More descriptive heading

VincentStimper · Jul 23, 2023 · d1a6af7 · d1a6af7
1 parent 20ee950
commit d1a6af7
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/examples/comparison_plan_rad_aff.ipynb b/examples/comparison_plan_rad_aff.ipynb
@@ -4,7 +4,13 @@
    "cell_type": "markdown",
    "metadata": {},
    "source": [
-    "# Flow training and visualization\n",
+    "# Comparison of Planar, Radial, and Affine Coupling Flows"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
     "In this notebook, we train normalizing flows to fit predefined prior distributions, testing their expressivity. The plots are generated to visualize the learned distributions for given layers $K$, and the training loss is plotted to compare the expressivity of different flows."
    ]
   },