def test_sequential_count_params():
input_dim = 20
num_units = 10
num_classes = 2
n = input_dim * num_units + num_units
n += num_units * num_units + num_units
n += num_units * num_classes + num_classes
model = Sequential()
model.add(Dense(num_units, input_shape=(input_dim,)))
model.add(Dense(num_units))
model.add(Dense(num_classes))
model.add(Activation('softmax'))
model.build()
assert(n == model.count_params())
model.compile('sgd', 'binary_crossentropy')
assert(n == model.count_params())
def test_sequential_count_params():
input_dim = 20
nb_units = 10
nb_classes = 2
n = input_dim * nb_units + nb_units
n += nb_units * nb_units + nb_units
n += nb_units * nb_classes + nb_classes
model = Sequential()
model.add(Dense(nb_units, input_shape=(input_dim,)))
model.add(Dense(nb_units))
model.add(Dense(nb_classes))
model.add(Activation("softmax"))
model.build()
assert n == model.count_params()
model.compile("sgd", "binary_crossentropy")
assert n == model.count_params()
