docs

ASEM000 · Sep 12, 2023 · 21a09c3 · 21a09c3
1 parent 241d7fa
commit 21a09c3
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 25 deletions.
diff --git a/docs/examples.rst b/docs/examples.rst
diff --git a/docs/index.rst b/docs/index.rst
@@ -70,14 +70,17 @@ Install from github::
     :maxdepth: 1
 
     notebooks/mental_model
-    notebooks/train_eval
-    notebooks/lazy_initialization
 
 .. toctree::
-    :caption: Examples
+    :caption: 📖 Guides
     :maxdepth: 1
 
-    examples
+    train_examples
+    notebooks/train_eval
+    notebooks/lazy_initialization
+    notebooks/train_mp
+    notebooks/checkpointing
+    notebooks/regularization
 
 
 .. currentmodule:: serket

diff --git a/docs/notebooks/checkpointing.ipynb b/docs/notebooks/checkpointing.ipynb
@@ -36,8 +36,7 @@
      "name": "stdout",
      "output_type": "stream",
      "text": [
-      "Original net\n",
-      "========================================\n",
+      "*Original net*\n",
       "Sequential(\n",
       "  layers=(\n",
       "    Linear(\n",
@@ -59,8 +58,7 @@
       "    )\n",
       "  )\n",
       ")\n",
-      "Loaded net\n",
-      "========================================\n",
+      "*Loaded net*\n",
       "Sequential(\n",
       "  layers=(\n",
       "    Linear(\n",
@@ -119,11 +117,9 @@
     "# 5) unmask the loaded tree\n",
     "loaded_net = sk.tree_unmask(loaded_net)\n",
     "\n",
-    "print(\"Original net\")\n",
-    "print(\"=\"*40)\n",
+    "print(\"*Original net*\")\n",
     "print(repr(net))\n",
-    "print(\"Loaded net\")\n",
-    "print(\"=\"*40)\n",
+    "print(\"*Loaded net*\")\n",
     "print(repr(loaded_net))"
    ]
   },