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.nas.space import AutoKSearchSpace from deephyper.nas.space.node import VariableNode from deephyper.nas.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.nas.space import AutoKSearchSpace from deephyper.nas.space.node import VariableNode from deephyper.nas.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)