folivo/PINA
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

add Poisson tutorial

Browse Source
This commit is contained in:
aivagnes
2022-03-21 10:48:59 +01:00
parent beb33f0da2
commit 36a3ad94bf
2 changed files with 3709 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
pina/plotter.py
View File

@@ -125,17 +125,17 @@ class Plotter:
pts.tensor[:, 1].reshape(res, res), pts.tensor[:, 1].reshape(res, res),
] ]
predicted_output = obj.model(pts) predicted_output = obj.model(pts)
predicted_output = predicted_output['ux'] predicted_output = predicted_output['u']
if hasattr(obj.problem, 'truth_solution'): if hasattr(obj.problem, 'truth_solution'):
truth_output = obj.problem.truth_solution(*pts.tensor.T).float() truth_output = obj.problem.truth_solution(*pts.tensor.T).float()
fig, axes = plt.subplots(nrows=1, ncols=3, figsize=(16, 6)) fig, axes = plt.subplots(nrows=1, ncols=3, figsize=(16, 6))
cb = getattr(axes[0], method)(*grids_container, predicted_output.tensor.reshape(res, res).detach()) cb = getattr(axes[0], method)(*grids_container, predicted_output.reshape(res, res).detach())
fig.colorbar(cb, ax=axes[0]) fig.colorbar(cb, ax=axes[0])
cb = getattr(axes[1], method)(*grids_container, truth_output.reshape(res, res).detach()) cb = getattr(axes[1], method)(*grids_container, truth_output.reshape(res, res).detach())
fig.colorbar(cb, ax=axes[1]) fig.colorbar(cb, ax=axes[1])
cb = getattr(axes[2], method)(*grids_container, (truth_output-predicted_output.tensor.float().flatten()).detach().reshape(res, res)) cb = getattr(axes[2], method)(*grids_container, (truth_output-predicted_output.float().flatten()).detach().reshape(res, res))
fig.colorbar(cb, ax=axes[2]) fig.colorbar(cb, ax=axes[2])
else: else:
fig, axes = plt.subplots(nrows=1, ncols=1, figsize=(8, 6)) fig, axes = plt.subplots(nrows=1, ncols=1, figsize=(8, 6))

3705
tutorials/Tutorial Poisson.ipynb vendored Normal file
View File

File diff suppressed because one or more lines are too long