def test_create_search_space(input_shape=(2, ), output_shape=(1, ), **kwargs):
struct = AutoKSearchSpace(input_shape, output_shape, regression=True)
vnode1 = VariableNode()
for _ in range(1, 11):
vnode1.add_op(Operation(layer=tf.keras.layers.Dense(10)))
struct.connect(struct.input_nodes[0], vnode1)
struct.set_ops([0])
struct.create_model()
def test_create_more_nodes(self):
from deephyper.search.nas.model.space import AutoKSearchSpace
from deephyper.search.nas.model.space.node import VariableNode
from deephyper.search.nas.model.space.op.op1d import Dense
struct = AutoKSearchSpace((5, ), (1, ), regression=True)
vnode1 = VariableNode()
struct.connect(struct.input_nodes[0], vnode1)
vnode1.add_op(Dense(10))
vnode2 = VariableNode()
vnode2.add_op(Dense(10))
struct.connect(vnode1, vnode2)
struct.set_ops([0, 0])
falias = 'test_auto_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(self):
from deephyper.search.nas.model.space import AutoKSearchSpace
from deephyper.search.nas.model.space.node import VariableNode
from deephyper.search.nas.model.space.op.op1d import Dense
struct = AutoKSearchSpace([(5, ), (5, )], (1, ), regression=True)
struct.set_ops([])
falias = 'test_auto_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)