Merge pull request #14 from aai-institute/feature/sine-notebook

Update notebook.
aai-institute · Dec 5, 2023 · 3f9bcc4 · 3f9bcc4
2 parents 5bc3c8b + 60b0352
commit 3f9bcc4
Showing 1 changed file with 3 additions and 12 deletions.
diff --git a/notebooks/sine.ipynb b/notebooks/sine.ipynb
@@ -6,8 +6,8 @@
    "source": [
     "# Sine Example\n",
     "\n",
-    "This example shows how to use a neural operator to learn the sine function in\n",
-    "a self-supervised manner."
+    "This example trains a neural operator on the sine function in a self-supervised \n",
+    "manner."
    ]
   },
   {
@@ -116,9 +116,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "fig, axs = plt.subplots(size // 4, 4, figsize=(16, 3 * size // 4))\n",
-    "if size // 4 == 1:\n",
-    "    axs = [axs]\n",
+    "fig, axs = plt.subplots(2, 4, figsize=(16, 6))\n",
     "\n",
     "for i in range(size):\n",
     "    ax = axs[i // 4][i % 4]\n",
@@ -145,13 +143,6 @@
     "plot_evaluation(model, dataset, obs, ax=ax)\n",
     "plot_observation(obs, ax=ax)"
    ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "metadata": {},
-   "outputs": [],
-   "source": []
   }
  ],
  "metadata": {