def docknet2(): docknet1 = Docknet() docknet1.add_input_layer(2) docknet1.add_dense_layer(3, 'relu') docknet1.add_dense_layer(1, 'sigmoid') docknet1.cost_function = 'cross_entropy' docknet1.initializer = DummyInitializer() docknet1.optimizer = AdamOptimizer() yield docknet1
def docknet1() -> Docknet: docknet1 = Docknet() docknet1.add_input_layer(2) docknet1.add_dense_layer(3, 'relu') docknet1.add_dense_layer(1, 'sigmoid') docknet1.cost_function = 'cross_entropy' docknet1.initializer = DummyInitializer() docknet1.optimizer = GradientDescentOptimizer() yield docknet1