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()