try more reproducibility

arjunsavel · Feb 29, 2024 · 1ccb826 · 1ccb826
1 parent 20ee63f
commit 1ccb826
Showing 1 changed file with 14 additions and 1 deletion.
diff --git a/docs/pages/fitting_nn.ipynb b/docs/pages/fitting_nn.ipynb
@@ -66,7 +66,12 @@
     "np.random.seed(seed)\n",
     "tf.random.set_seed(seed)\n",
     "os.environ[\"PYTHONHASHSEED\"] = str(seed)\n",
-    "random.seed(seed)"
+    "random.seed(seed)\n",
+    "\n",
+    "keras.utils.set_random_seed(seed)\n",
+    "\n",
+    "# note that setting this flag helps with reproducibility, but it may result in a performance hit.\n",
+    "tf.config.experimental.enable_op_determinism()"
    ]
   },
   {
@@ -431,6 +436,14 @@
    "metadata": {},
    "outputs": [],
    "source": []
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "4e8f6df0-a216-45a1-ba14-1cdc4755b724",
+   "metadata": {},
+   "outputs": [],
+   "source": []
   }
  ],
  "metadata": {