예제 #1
0
 def test_returns_layer(self):
     layer = Layer(size=3,
                   prev_size=2,
                   activation=activation_functions.Rectifier)
     next_layer = layer.create_next_layer(
         size=5, activation=activation_functions.Rectifier)
     self.assertIsInstance(next_layer, Layer)
예제 #2
0
    def test_created_layer_has_correct_activation_function(self):
        layer = Layer(size=3,
                      prev_size=2,
                      activation=activation_functions.Rectifier)
        next_layer = layer.create_next_layer(
            size=5, activation=activation_functions.Rectifier)
        self.assertEqual(next_layer.get_activation(),
                         activation_functions.Rectifier)

        next_layer = next_layer.create_next_layer(
            size=5, activation=activation_functions.Sigmoid)
        self.assertEqual(next_layer.get_activation(),
                         activation_functions.Sigmoid)
예제 #3
0
    def test_created_layer_has_correct_parameters(self):
        layer = Layer(size=3,
                      prev_size=2,
                      activation=activation_functions.Rectifier)
        next_layer = layer.create_next_layer(
            size=5, activation=activation_functions.Rectifier)
        self.assertTupleEqual(next_layer.biases().shape, (5, ))
        self.assertTupleEqual(next_layer.weights().shape, (5, 3))

        next_layer = next_layer.create_next_layer(
            size=4, activation=activation_functions.Rectifier)
        self.assertTupleEqual(next_layer.biases().shape, (4, ))
        self.assertTupleEqual(next_layer.weights().shape, (4, 5))