add badges

ASEM000 · Jan 29, 2024 · 6fe38d8 · 6fe38d8
1 parent fae40f3
commit 6fe38d8
Show file tree

Hide file tree

Showing 7 changed files with 7 additions and 8 deletions.
diff --git a/docs/notebooks/train_bilstm.ipynb b/docs/notebooks/train_bilstm.ipynb
@@ -4,7 +4,7 @@
    "cell_type": "markdown",
    "metadata": {},
    "source": [
-    "# ![](https://img.shields.io/badge/time%20series-FFF2DB) Train bidirectional-`LSTM`"
+    "# ![](https://img.shields.io/badge/time%20series-6c4ce2) Train bidirectional-`LSTM`"
    ]
   },
   {

diff --git a/docs/notebooks/train_convlstm.ipynb b/docs/notebooks/train_convlstm.ipynb
@@ -6,7 +6,7 @@
         "id": "PJyyxjLA2A62"
       },
       "source": [
-        "# Train Convolutional `LSTM`\n",
+        "# ![](https://img.shields.io/badge/vision-2f5ada)  ![](https://img.shields.io/badge/time%20series-6c4ce2)  Train Convolutional `LSTM`\n",
         "\n",
         "\n",
         "In this notebook, a simple next frame video prediction using convolutional `LSTM` is demonstrated."

diff --git a/docs/notebooks/train_fourier_features_network.ipynb b/docs/notebooks/train_fourier_features_network.ipynb
@@ -5,7 +5,7 @@
    "id": "b2b68cc9-825c-490f-9476-7b43acec0eaa",
    "metadata": {},
    "source": [
-    "# Train fourier features network\n",
+    "# ![](https://img.shields.io/badge/vision-2f5ada)  Train fourier features network\n",
     "\n",
     "_Fourier Features Let Networks Learn High Frequency Functions in Low Dimensional Domains_\n",
     "\n",
@@ -49,7 +49,6 @@
     "import matplotlib.pyplot as plt  # visualization\n",
     "import imageio\n",
     "from typing_extensions import Annotated\n",
-    "import numpy as np\n",
     "import serket as sk\n",
     "import optax\n",
     "import functools as ft"

diff --git a/docs/notebooks/train_mnist.ipynb b/docs/notebooks/train_mnist.ipynb
@@ -6,7 +6,7 @@
         "id": "k0i-yEycZkF-"
       },
       "source": [
-        "# Train `MNIST`"
+        "# ![](https://img.shields.io/badge/vision-2f5ada) Train `MNIST`"
       ]
     },
     {

diff --git a/docs/notebooks/train_pinn_burgers.ipynb b/docs/notebooks/train_pinn_burgers.ipynb
@@ -4,7 +4,7 @@
    "cell_type": "markdown",
    "metadata": {},
    "source": [
-    "# Train `PINN`\n",
+    "# ![](https://img.shields.io/badge/science-9932cc) Train `PINN`\n",
     "\n",
     "In this example, [physics informed neural network (PINN)](https://maziarraissi.github.io/PINNs/) technique is used to train [burgers equation](https://en.wikipedia.org/wiki/Burgers%27_equation).\n",
     "\n",

diff --git a/docs/notebooks/train_transformer.ipynb b/docs/notebooks/train_transformer.ipynb
@@ -4,7 +4,7 @@
    "cell_type": "markdown",
    "metadata": {},
    "source": [
-    "# Train transformer\n",
+    "# ![](https://img.shields.io/badge/language-cb2888) Train transformer\n",
     "\n",
     "In this guide,  a simple transformer model is trained from scratch to perform machine translation as detailed in [Attention is all you need](https://arxiv.org/abs/1706.03762) from Arabic to English. But you are free to choose any language pair from the configuration."
    ]

diff --git a/docs/notebooks/train_unet.ipynb b/docs/notebooks/train_unet.ipynb
@@ -4,7 +4,7 @@
    "cell_type": "markdown",
    "metadata": {},
    "source": [
-    "# Train `UNet` segmenter\n",
+    "# ![](https://img.shields.io/badge/vision-2f5ada) Train `UNet` segmenter\n",
     "\n",
     "\n",
     "\n",