updating dense notebooks

FNALLPC · Jan 14, 2025 · 91c97c1 · 91c97c1
1 parent df3ee93
commit 91c97c1
Show file tree

Hide file tree

Showing 5 changed files with 793 additions and 2 deletions.
diff --git a/machine-learning-das/_toc.yml b/machine-learning-das/_toc.yml
@@ -19,7 +19,8 @@ parts:
       sections:
         - file: notebooks/3.1-dense-keras
         - file: notebooks/3.2-dense-pytorch
-        - file: notebooks/3.3-dense-bayesian-optimization
+        - file: notebooks/3.3-dense-keras-JetTagging.ipynb
+        - file: notebooks/3.4-dense-bayesian-optimization.ipynb
     - file: notebooks/4-conv2d
     - file: notebooks/5-gnn-cora
     - file: notebooks/6-vae-mnist

diff --git a/machine-learning-das/notebooks/3.1-dense-keras.ipynb b/machine-learning-das/notebooks/3.1-dense-keras.ipynb
@@ -278,7 +278,7 @@
     "from tensorflow.keras.callbacks import ModelCheckpoint\n",
     "\n",
     "model_checkpoint = ModelCheckpoint(\n",
-    "    \"dense_model.h5\",\n",
+    "    \"dense_model.keras\",\n",
     "    monitor=\"val_loss\",\n",
     "    verbose=0,\n",
     "    save_best_only=True,\n",