Merge pull request #4 from levimcclenny/main

add model.save and model.load_model

Author: levimcclenny

tensordiffeq/models.py

@@ -180,6 +180,11 @@ def predict(self, X_star):
         f_u_star = self.f_model(self.u_model, *X_star)
         return u_star.numpy(), f_u_star.numpy()
 
+    def save(self, path):
+        self.u_model.save(path)
+
+    def load_model(self, path):
+        self.u_model = tf.keras.models.load_model(path)
 
 # WIP
 # TODO Distributed Discovery Model