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

minor changes

Browse Source
This commit is contained in:
Nicola Demo
2022-02-11 17:24:12 +01:00
parent 1483746b45
commit 1812ddb8d9
5 changed files with 12 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
examples/problems/burgers.py
View File

@@ -15,7 +15,8 @@ class Burgers1D(TimeDependentProblem, SpatialProblem):
def burger_equation(input_, output_): def burger_equation(input_, output_):
grad_u = grad(output_['u'], input_) grad_u = grad(output_['u'], input_)
grad_x, grad_t = grad(output_['u'], input_).T grad_x = grad_u['x']
grad_t = grad_u['t']
gradgrad_u_x = grad(grad_u['x'], input_) gradgrad_u_x = grad(grad_u['x'], input_)
return ( return (
grad_u['t'] + output_['u']*grad_u['x'] - grad_u['t'] + output_['u']*grad_u['x'] -

2
examples/run_burgers.py
View File

@@ -37,7 +37,7 @@ if __name__ == "__main__":
output_variables=burgers_problem.output_variables, output_variables=burgers_problem.output_variables,
input_variables=burgers_problem.input_variables, input_variables=burgers_problem.input_variables,
func=Softplus, func=Softplus,
extra_features=feat extra_features=feat,
) )
pinn = PINN( pinn = PINN(

15
examples/run_poisson.py
View File

@@ -1,17 +1,14 @@
import argparse
import sys import sys
import numpy as np import numpy as np
import torch import torch
import argparse
from pina import PINN
from pina.ppinn import ParametricPINN as pPINN
from pina.label_tensor import LabelTensor
from torch.nn import ReLU, Tanh, Softplus from torch.nn import ReLU, Tanh, Softplus
from problems.poisson2D import Poisson2D
from pina.deep_feed_forward import DeepFeedForward
from pina import PINN, LabelTensor, Plotter
from pina.model import FeedForward
from pina.adaptive_functions import AdaptiveSin, AdaptiveCos, AdaptiveTanh from pina.adaptive_functions import AdaptiveSin, AdaptiveCos, AdaptiveTanh
from problems.poisson import Poisson
from pina import Plotter
class myFeature(torch.nn.Module): class myFeature(torch.nn.Module):
""" """
@@ -36,8 +33,8 @@ if __name__ == "__main__":
feat = [myFeature()] if args.features else [] feat = [myFeature()] if args.features else []
poisson_problem = Poisson2D() poisson_problem = Poisson()
model = DeepFeedForward( model = FeedForward(
layers=[10, 10], layers=[10, 10],
output_variables=poisson_problem.output_variables, output_variables=poisson_problem.output_variables,
input_variables=poisson_problem.input_variables, input_variables=poisson_problem.input_variables,

2
pina/plotter.py
View File

@@ -1,6 +1,6 @@
""" Module for plotting. """ """ Module for plotting. """
import matplotlib import matplotlib
matplotlib.use('Qt5Agg') #matplotlib.use('Qt5Agg')
import matplotlib.pyplot as plt import matplotlib.pyplot as plt
import numpy as np import numpy as np
import torch import torch

3
setup.py
View File

@@ -14,7 +14,8 @@ VERSION = meta['__version__']
KEYWORDS = 'physics-informed neural-network' KEYWORDS = 'physics-informed neural-network'
REQUIRED = [ REQUIRED = [
'future', 'numpy', 'matplotlib', 'torch' 'future', 'numpy', 'matplotlib', 'torch', 'mpmath'
] ]
EXTRAS = { EXTRAS = {