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

variable name fix FeedForward model

Browse Source
This commit is contained in:
Dario Coscia
2023-09-13 12:48:55 +02:00
committed by Nicola Demo
parent 17464ceca9
commit 5a4c114d48
3 changed files with 29 additions and 29 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
tests/test_model/test_deeponet.py
View File

@@ -11,8 +11,8 @@ input_ = LabelTensor(data, input_vars)
def test_constructor():
branch_net = FeedForward(input_dimensons=1, output_dimensions=10)
trunk_net = FeedForward(input_dimensons=2, output_dimensions=10)
branch_net = FeedForward(input_dimensions=1, output_dimensions=10)
trunk_net = FeedForward(input_dimensions=2, output_dimensions=10)
DeepONet(branch_net=branch_net,
trunk_net=trunk_net,
input_indeces_branch_net=['a'],
@@ -22,8 +22,8 @@ def test_constructor():
def test_constructor_fails_when_invalid_inner_layer_size():
branch_net = FeedForward(input_dimensons=1, output_dimensions=10)
trunk_net = FeedForward(input_dimensons=2, output_dimensions=8)
branch_net = FeedForward(input_dimensions=1, output_dimensions=10)
trunk_net = FeedForward(input_dimensions=2, output_dimensions=8)
with pytest.raises(ValueError):
DeepONet(branch_net=branch_net,
trunk_net=trunk_net,
@@ -33,8 +33,8 @@ def test_constructor_fails_when_invalid_inner_layer_size():
aggregator='*')
def test_forward_extract_str():
branch_net = FeedForward(input_dimensons=1, output_dimensions=10)
trunk_net = FeedForward(input_dimensons=2, output_dimensions=10)
branch_net = FeedForward(input_dimensions=1, output_dimensions=10)
trunk_net = FeedForward(input_dimensions=2, output_dimensions=10)
model = DeepONet(branch_net=branch_net,
trunk_net=trunk_net,
input_indeces_branch_net=['a'],
@@ -44,8 +44,8 @@ def test_forward_extract_str():
model(input_)
def test_forward_extract_int():
branch_net = FeedForward(input_dimensons=1, output_dimensions=10)
trunk_net = FeedForward(input_dimensons=2, output_dimensions=10)
branch_net = FeedForward(input_dimensions=1, output_dimensions=10)
trunk_net = FeedForward(input_dimensions=2, output_dimensions=10)
model = DeepONet(branch_net=branch_net,
trunk_net=trunk_net,
input_indeces_branch_net=[0],
@@ -55,8 +55,8 @@ def test_forward_extract_int():
model(data)
def test_forward_extract_str_wrong():
branch_net = FeedForward(input_dimensons=1, output_dimensions=10)
trunk_net = FeedForward(input_dimensons=2, output_dimensions=10)
branch_net = FeedForward(input_dimensions=1, output_dimensions=10)
trunk_net = FeedForward(input_dimensions=2, output_dimensions=10)
model = DeepONet(branch_net=branch_net,
trunk_net=trunk_net,
input_indeces_branch_net=['a'],

30
tests/test_model/test_mionet.py
View File

@@ -11,9 +11,9 @@ input_ = LabelTensor(data, input_vars)
def test_constructor():
branch_net1 = FeedForward(input_dimensons=1, output_dimensions=10)
branch_net2 = FeedForward(input_dimensons=2, output_dimensions=10)
trunk_net = FeedForward(input_dimensons=1, output_dimensions=10)
branch_net1 = FeedForward(input_dimensions=1, output_dimensions=10)
branch_net2 = FeedForward(input_dimensions=2, output_dimensions=10)
trunk_net = FeedForward(input_dimensions=1, output_dimensions=10)
networks = {branch_net1 : ['x'],
branch_net2 : ['x', 'y'],
trunk_net : ['z']}
@@ -23,9 +23,9 @@ def test_constructor():
def test_constructor_fails_when_invalid_inner_layer_size():
branch_net1 = FeedForward(input_dimensons=1, output_dimensions=10)
branch_net2 = FeedForward(input_dimensons=2, output_dimensions=10)
trunk_net = FeedForward(input_dimensons=1, output_dimensions=12)
branch_net1 = FeedForward(input_dimensions=1, output_dimensions=10)
branch_net2 = FeedForward(input_dimensions=2, output_dimensions=10)
trunk_net = FeedForward(input_dimensions=1, output_dimensions=12)
networks = {branch_net1 : ['x'],
branch_net2 : ['x', 'y'],
trunk_net : ['z']}
@@ -35,9 +35,9 @@ def test_constructor_fails_when_invalid_inner_layer_size():
aggregator='*')
def test_forward_extract_str():
branch_net1 = FeedForward(input_dimensons=1, output_dimensions=10)
branch_net2 = FeedForward(input_dimensons=1, output_dimensions=10)
trunk_net = FeedForward(input_dimensons=1, output_dimensions=10)
branch_net1 = FeedForward(input_dimensions=1, output_dimensions=10)
branch_net2 = FeedForward(input_dimensions=1, output_dimensions=10)
trunk_net = FeedForward(input_dimensions=1, output_dimensions=10)
networks = {branch_net1 : ['a'],
branch_net2 : ['b'],
trunk_net : ['c']}
@@ -47,9 +47,9 @@ def test_forward_extract_str():
model(input_)
def test_forward_extract_int():
branch_net1 = FeedForward(input_dimensons=1, output_dimensions=10)
branch_net2 = FeedForward(input_dimensons=1, output_dimensions=10)
trunk_net = FeedForward(input_dimensons=1, output_dimensions=10)
branch_net1 = FeedForward(input_dimensions=1, output_dimensions=10)
branch_net2 = FeedForward(input_dimensions=1, output_dimensions=10)
trunk_net = FeedForward(input_dimensions=1, output_dimensions=10)
networks = {branch_net1 : [0],
branch_net2 : [1],
trunk_net : [2]}
@@ -59,9 +59,9 @@ def test_forward_extract_int():
model(data)
def test_forward_extract_str_wrong():
branch_net1 = FeedForward(input_dimensons=1, output_dimensions=10)
branch_net2 = FeedForward(input_dimensons=1, output_dimensions=10)
trunk_net = FeedForward(input_dimensons=1, output_dimensions=10)
branch_net1 = FeedForward(input_dimensions=1, output_dimensions=10)
branch_net2 = FeedForward(input_dimensions=1, output_dimensions=10)
trunk_net = FeedForward(input_dimensions=1, output_dimensions=10)
networks = {branch_net1 : ['a'],
branch_net2 : ['b'],
trunk_net : ['c']}