add units for PINN forward examples

chaobrain · Dec 17, 2024 · a006228 · a006228
1 parent f837ccc
commit a006228
Show file tree

Hide file tree

Showing 17 changed files with 1,917 additions and 256 deletions.
diff --git a/docs/examples-pinn-forward/Beltrami_flow.ipynb b/docs/examples-pinn-forward/Beltrami_flow.ipynb
diff --git a/docs/examples-pinn-forward/Euler_beam.ipynb b/docs/examples-pinn-forward/Euler_beam.ipynb
diff --git a/docs/examples-pinn-forward/Helmholtz_Dirichlet_2d.ipynb b/docs/examples-pinn-forward/Helmholtz_Dirichlet_2d.ipynb
diff --git a/docs/examples-pinn-forward/Klein_Gordon.ipynb b/docs/examples-pinn-forward/Klein_Gordon.ipynb
@@ -34,6 +34,13 @@
     "\n",
     "The reference solution is $y(x, t) = x\\cos(t)$."
    ]
+  },
+  {
+   "metadata": {},
+   "cell_type": "code",
+   "outputs": [],
+   "execution_count": null,
+   "source": ""
   }
  ],
  "metadata": {

diff --git a/docs/examples-pinn-forward/Kovasznay_flow.ipynb b/docs/examples-pinn-forward/Kovasznay_flow.ipynb
@@ -26,6 +26,13 @@
     "\n",
     "The reference solution is $u = 1 - e^{\\lambda x} \\cos(2\\pi y), v = \\frac{\\lambda}{2\\pi}e^{\\lambda x} \\sin(2\\pi x)$, $p =\\frac{1}{2}(1 - e^{2\\lambda x})$, where $\\lambda = \\frac{1}{2\\nu}-\\sqrt{\\frac{1}{4\\nu^2}+4\\pi^2}$."
    ]
+  },
+  {
+   "metadata": {},
+   "cell_type": "code",
+   "outputs": [],
+   "execution_count": null,
+   "source": ""
   }
  ],
  "metadata": {

diff --git a/docs/examples-pinn-forward/Kovasznay_flow.py b/docs/examples-pinn-forward/Kovasznay_flow.py
@@ -5,6 +5,7 @@
 
 import pinnx
 
+
 Re = 20
 nu = 1 / Re
 l = 1 / (2 * nu) - u.math.sqrt(1 / (4 * nu ** 2) + 4 * u.math.pi ** 2)
@@ -33,7 +34,6 @@ def pde(x, y):
     continuity = u_vel_x + v_vel_y
 
     return momentum_x, momentum_y, continuity
-    # return {'momentum_u': momentum_x, 'momentum_v': momentum_y, 'continuity': continuity}
 
 
 def bc_func(x):

diff --git a/docs/examples-pinn-forward/diffusion_1d.ipynb b/docs/examples-pinn-forward/diffusion_1d.ipynb
diff --git a/docs/examples-pinn-forward/elasticity_plate.py b/docs/examples-pinn-forward/elasticity_plate.py
@@ -86,8 +86,7 @@ def fx(x):
         -lmbd * (
         4 * u.math.pi ** 2 * u.math.cos(2 * u.math.pi * x['x']) * u.math.sin(u.math.pi * x['y'])
         - Q * x['y'] ** 3 * u.math.pi * u.math.cos(u.math.pi * x['x'])
-    )
-        -
+    ) -
         mu * (
             u.math.pi ** 2 * u.math.cos(2 * u.math.pi * x['x']) * u.math.sin(u.math.pi * x['y'])
             - Q * x['y'] ** 3 * u.math.pi * u.math.cos(u.math.pi * x['x'])
@@ -112,7 +111,6 @@ def fy(x):
     )
 
 
-
 def pde(x, y):
     jacobian = net.jacobian(x)
 
@@ -127,7 +125,6 @@ def pde(x, y):
     Sxx_x = jacobian["s"]["x"]
     Syy_y = jacobian["c"]["y"]
     Sxy_x = jacobian["e"]["x"]
-    # Sxy_x = jacobian(f, x, i=4, j=0)
     Sxy_y = jacobian['e']['y']
 
     momentum_x = Sxx_x + Sxy_y - fx(x)
@@ -179,4 +176,3 @@ def pde(x, y):
 trainer = pinnx.Trainer(data)
 trainer.compile(bst.optim.Adam(0.001), metrics=["l2 relative error"]).train(iterations=1000)
 trainer.saveplot(issave=True, isplot=True)
-
diff --git a/docs/unit-examples-forward.rst b/docs/unit-examples-forward.rst
@@ -4,6 +4,10 @@ PINN Forward Examples
 .. toctree::
     :maxdepth: 1
 
+    unit-examples-forward/Beltrami_flow.ipynb
+    unit-examples-forward/diffusion_1d.ipynb
+    unit-examples-forward/Euler_beam.ipynb
+    unit-examples-forward/Helmholtz_Dirichlet_2d.ipynb
     unit-examples-forward/burgers.ipynb
     unit-examples-forward/Burgers_RAR.ipynb
     unit-examples-forward/heat.ipynb