def test_count_parameters(self): connection = layers.join( layers.Input(10), layers.Sigmoid(5), layers.Sigmoid(2), ) n_parameters = layers.count_parameters(connection) self.assertEqual(n_parameters, (10 * 5 + 5) + (5 * 2 + 2))
def test_count_parameters_single_layer(self): hidden_layer = layers.Sigmoid(5) layers.join( layers.Input(10), hidden_layer, layers.Sigmoid(2), ) n_parameters = layers.count_parameters(hidden_layer) self.assertEqual(n_parameters, 10 * 5 + 5)