fixup! Add support for TensorFlow 2.16

nengo · Apr 5, 2024 · abf92c4 · abf92c4
1 parent d2ac24f
commit abf92c4
Show file tree

Hide file tree

Showing 4 changed files with 6 additions and 3 deletions.
diff --git a/docs/examples/psMNIST-training.png b/docs/examples/psMNIST-training.png
diff --git a/docs/examples/psMNIST-weights.hdf5 b/docs/examples/psMNIST-weights.hdf5
diff --git a/docs/examples/psMNIST.ipynb b/docs/examples/psMNIST.ipynb
@@ -317,10 +317,13 @@
     "batch_size = 100\n",
     "epochs = 10\n",
     "\n",
-    "saved_weights_fname = \"./psMNIST-weights.hdf5\"\n",
+    "saved_model_fname = \"./psMNIST.keras\"\n",
     "callbacks = [\n",
     "    keras.callbacks.ModelCheckpoint(\n",
-    "        filepath=saved_weights_fname, monitor=\"val_loss\", verbose=1, save_best_only=True\n",
+    "        filepath=saved_model_fname,\n",
+    "        monitor=\"val_accuracy\",\n",
+    "        verbose=1,\n",
+    "        save_best_only=True,\n",
     "    ),\n",
     "]\n",
     "\n",
@@ -394,7 +397,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "model.load_weights(saved_weights_fname)\n",
+    "model.load_weights(saved_model_fname)\n",
     "accuracy = model.evaluate(X_test, Y_test)[1] * 100\n",
     "print(f\"Test accuracy: {round(accuracy, 2):0.2f}%\")"
    ]

diff --git a/docs/examples/psMNIST.keras b/docs/examples/psMNIST.keras