Exemple #1
0
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
Exemple #2
0
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