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

fix bug network

Browse Source
This commit is contained in:
Dario Coscia
2023-11-13 12:25:40 +01:00
committed by Nicola Demo
parent ee39b39805
commit a9f14ac323
6 changed files with 127 additions and 80 deletions
Download Patch File Download Diff File Expand all files Collapse all files

62
tutorials/tutorial5/tutorial.ipynb vendored
View File

@@ -19,7 +19,7 @@
},
{
"cell_type": "code",
"execution_count": 11,
"execution_count": 1,
"id": "5f2744dc",
"metadata": {},
"outputs": [],
@@ -28,8 +28,7 @@
"from scipy import io\n",
"import torch\n",
"from pina.model import FNO, FeedForward # let's import some models\n",
"from pina import Condition\n",
"from pina import LabelTensor\n",
"from pina import Condition, LabelTensor\n",
"from pina.solvers import SupervisedSolver\n",
"from pina.trainer import Trainer\n",
"from pina.problem import AbstractProblem\n",
@@ -63,10 +62,10 @@
"data = io.loadmat(\"Data_Darcy.mat\")\n",
"\n",
"# extract data (we use only 100 data for train)\n",
"k_train = torch.tensor(data['k_train'], dtype=torch.float).unsqueeze(-1)\n",
"u_train = torch.tensor(data['u_train'], dtype=torch.float).unsqueeze(-1)\n",
"k_test = torch.tensor(data['k_test'], dtype=torch.float).unsqueeze(-1)\n",
"u_test= torch.tensor(data['u_test'], dtype=torch.float).unsqueeze(-1)\n",
"k_train = LabelTensor(torch.tensor(data['k_train'], dtype=torch.float).unsqueeze(-1), ['u0'])\n",
"u_train = LabelTensor(torch.tensor(data['u_train'], dtype=torch.float).unsqueeze(-1), ['u'])\n",
"k_test = LabelTensor(torch.tensor(data['k_test'], dtype=torch.float).unsqueeze(-1), ['u0'])\n",
"u_test= LabelTensor(torch.tensor(data['u_test'], dtype=torch.float).unsqueeze(-1), ['u'])\n",
"x = torch.tensor(data['x'], dtype=torch.float)[0]\n",
"y = torch.tensor(data['y'], dtype=torch.float)[0]"
]
@@ -116,16 +115,16 @@
},
{
"cell_type": "code",
"execution_count": 14,
"execution_count": 17,
"id": "8b27d283",
"metadata": {},
"outputs": [],
"source": [
"class NeuralOperatorSolver(AbstractProblem):\n",
" input_variables = ['u_0']\n",
" output_variables = ['u']\n",
" conditions = {'data' : Condition(input_points=LabelTensor(k_train, input_variables), \n",
" output_points=LabelTensor(u_train, output_variables))}\n",
" input_variables = k_train.labels\n",
" output_variables = u_train.labels\n",
" conditions = {'data' : Condition(input_points=k_train, \n",
" output_points=u_train)}\n",
"\n",
"# make problem\n",
"problem = NeuralOperatorSolver()"
@@ -143,7 +142,7 @@
},
{
"cell_type": "code",
"execution_count": 15,
"execution_count": 18,
"id": "e34f18b0",
"metadata": {},
"outputs": [
@@ -161,7 +160,7 @@
"name": "stdout",
"output_type": "stream",
"text": [
"Epoch 9: : 100it [00:00, 383.36it/s, v_num=36, mean_loss=0.108]"
"Epoch 9: : 100it [00:00, 357.28it/s, v_num=1, mean_loss=0.108]"
]
},
{
@@ -175,7 +174,7 @@
"name": "stdout",
"output_type": "stream",
"text": [
"Epoch 9: : 100it [00:00, 380.57it/s, v_num=36, mean_loss=0.108]\n"
"Epoch 9: : 100it [00:00, 354.81it/s, v_num=1, mean_loss=0.108]\n"
]
}
],
@@ -202,7 +201,7 @@
},
{
"cell_type": "code",
"execution_count": 16,
"execution_count": 19,
"id": "0e2a6aa4",
"metadata": {},
"outputs": [
@@ -222,10 +221,10 @@
"metric_err = LpLoss(relative=True)\n",
"\n",
"\n",
"err = float(metric_err(u_train.squeeze(-1), solver.models[0](k_train).squeeze(-1)).mean())*100\n",
"err = float(metric_err(u_train.squeeze(-1), solver.neural_net(k_train).squeeze(-1)).mean())*100\n",
"print(f'Final error training {err:.2f}%')\n",
"\n",
"err = float(metric_err(u_test.squeeze(-1), solver.models[0](k_test).squeeze(-1)).mean())*100\n",
"err = float(metric_err(u_test.squeeze(-1), solver.neural_net(k_test).squeeze(-1)).mean())*100\n",
"print(f'Final error testing {err:.2f}%')"
]
},
@@ -241,7 +240,7 @@
},
{
"cell_type": "code",
"execution_count": 17,
"execution_count": 24,
"id": "9af523a5",
"metadata": {},
"outputs": [
@@ -259,7 +258,14 @@
"name": "stdout",
"output_type": "stream",
"text": [
"Epoch 9: : 100it [00:04, 22.13it/s, v_num=37, mean_loss=0.000952]"
"Epoch 0: : 0it [00:00, ?it/s]"
]
},
{
"name": "stdout",
"output_type": "stream",
"text": [
"Epoch 9: : 100it [00:02, 47.76it/s, v_num=4, mean_loss=0.00106] "
]
},
{
@@ -273,7 +279,7 @@
"name": "stdout",
"output_type": "stream",
"text": [
"Epoch 9: : 100it [00:04, 22.07it/s, v_num=37, mean_loss=0.000952]\n"
"Epoch 9: : 100it [00:02, 47.65it/s, v_num=4, mean_loss=0.00106]\n"
]
}
],
@@ -283,10 +289,10 @@
"projecting_net = torch.nn.Linear(24, 1)\n",
"model = FNO(lifting_net=lifting_net,\n",
" projecting_net=projecting_net,\n",
" n_modes=16,\n",
" n_modes=8,\n",
" dimensions=2,\n",
" inner_size=24,\n",
" padding=11)\n",
" padding=8)\n",
"\n",
"\n",
"# make solver\n",
@@ -307,7 +313,7 @@
},
{
"cell_type": "code",
"execution_count": 18,
"execution_count": 25,
"id": "58e2db89",
"metadata": {},
"outputs": [
@@ -315,16 +321,16 @@
"name": "stdout",
"output_type": "stream",
"text": [
"Final error training 4.45%\n",
"Final error testing 4.91%\n"
"Final error training 4.83%\n",
"Final error testing 5.16%\n"
]
}
],
"source": [
"err = float(metric_err(u_train.squeeze(-1), solver.models[0](k_train).squeeze(-1)).mean())*100\n",
"err = float(metric_err(u_train.squeeze(-1), solver.neural_net(k_train).squeeze(-1)).mean())*100\n",
"print(f'Final error training {err:.2f}%')\n",
"\n",
"err = float(metric_err(u_test.squeeze(-1), solver.models[0](k_test).squeeze(-1)).mean())*100\n",
"err = float(metric_err(u_test.squeeze(-1), solver.neural_net(k_test).squeeze(-1)).mean())*100\n",
"print(f'Final error testing {err:.2f}%')"
]
},