Merge pull request #153 from IamShubhamGupto/fix/gp-lorenz

fix GP deprecation with tf-keras
m2lines · Jun 6, 2024 · 1980cfd · 1980cfd
2 parents da2f5de + 1425631
commit 1980cfd
Show file tree

Hide file tree

Showing 4 changed files with 6 additions and 9 deletions.
diff --git a/notebooks/GP_lorenz96.ipynb b/notebooks/GP_lorenz96.ipynb
@@ -19,6 +19,9 @@
     "import gpflow\n",
     "from L96_model import L96, L96_eq1_xdot, integrate_L96_2t\n",
     "from L96_model import EulerFwd, RK2, RK4\n",
+    "import os\n",
+    "\n",
+    "os.environ[\"TF_USE_LEGACY_KERAS\"] = \"True\"\n",
     "\n",
     "time_method = RK4"
    ]
@@ -335,7 +338,7 @@
     "    train_iter = iter(train_dataset.batch(batch_size))\n",
     "    training_loss = model.training_loss_closure(train_iter, compile=True)\n",
     "\n",
-    "    optimizer = tf.optimizers.Adam()\n",
+    "    optimizer = tf.keras.optimizers.legacy.Adam()\n",
     "\n",
     "    @tf.function\n",
     "    def optimization_step(trainable_variables):\n",
@@ -893,13 +896,6 @@
     "plt.legend(frameon=False, fontsize=7)\n",
     "plt.show()"
    ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "metadata": {},
-   "outputs": [],
-   "source": []
   }
  ],
  "metadata": {
@@ -918,7 +914,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.10.12"
+   "version": "3.9.19"
   }
  },
  "nbformat": 4,

diff --git a/notebooks/figs/Burgers_simulation.gif b/notebooks/figs/Burgers_simulation.gif
diff --git a/notebooks/figs/L96_one_variable.gif b/notebooks/figs/L96_one_variable.gif
diff --git a/requirements.txt b/requirements.txt
@@ -4,3 +4,4 @@ pysindy
 torch-lr-finder
 gplearn
 gpflow
+tf-keras