def test_create_more_nodes(self):
from deephyper.search.nas.model.space import KSearchSpace
from deephyper.search.nas.model.space.node import VariableNode
from deephyper.search.nas.model.space.op.op1d import Dense
struct = KSearchSpace((5, ), (1, ))
vnode1 = VariableNode()
struct.connect(struct.input_nodes[0], vnode1)
vnode1.add_op(Dense(10))
vnode2 = VariableNode()
vnode2.add_op(Dense(1))
struct.connect(vnode1, vnode2)
struct.set_ops([0, 0])
falias = 'test_keras_search_spaceure'
struct.draw_graphviz(f'{falias}.dot')
model = struct.create_model()
from tensorflow.keras.utils import plot_model
plot_model(model, to_file=f'{falias}.png', show_shapes=True)
def test_create_multiple_inputs_with_wrong_output_shape(self):
from deephyper.search.nas.model.space import KSearchSpace
from deephyper.search.nas.model.space.node import VariableNode
from deephyper.search.nas.model.space.op.op1d import Dense
struct = KSearchSpace([(5, ), (5, )], (1, ))
struct.set_ops([])
with pytest.raises(WrongOutputShape):
struct.create_model()
def test_create_one_vnode_with_wrong_output_shape(self):
from deephyper.search.nas.model.space import KSearchSpace
struct = KSearchSpace((5, ), (1, ))
from deephyper.search.nas.model.space.node import VariableNode
vnode = VariableNode()
struct.connect(struct.input_nodes[0], vnode)
from deephyper.search.nas.model.space.op.op1d import Dense
vnode.add_op(Dense(10))
struct.set_ops([0])
with pytest.raises(WrongOutputShape):
struct.create_model()
def test_create_multiple_inputs_with_one_vnode(self):
from deephyper.search.nas.model.space import KSearchSpace
from deephyper.search.nas.model.space.node import VariableNode, ConstantNode
from deephyper.search.nas.model.space.op.op1d import Dense
from deephyper.search.nas.model.space.op.merge import Concatenate
struct = KSearchSpace([(5, ), (5, )], (1, ))
merge = ConstantNode()
merge.set_op(Concatenate(struct, struct.input_nodes))
vnode1 = VariableNode()
struct.connect(merge, vnode1)
vnode1.add_op(Dense(1))
struct.set_ops([0])
struct.create_model()