Подтвердить что ты не робот

Как найти количество параметров модели keras?

Для сети Feedforward (FFN) легко вычислить количество параметров. Учитывая CNN, LSTM и т.д. Есть быстрый способ найти количество параметров в модели keras?

Ответ 1

Модели и слои имеют специальный метод для этой цели:

model.count_params()

Кроме того, чтобы получить краткое описание размеров и параметров каждого слоя, вы можете найти следующий метод

model.summary()

Ответ 2

import keras.backend as K

def size(model): # Compute number of params in a model (the actual number of floats)
    return sum([np.prod(K.get_value(w).shape) for w in model.trainable_weights])