如何查找keras模型的參數數量？

Question

對於前饋網絡（FFN），很容易計算參數的數量。給定一個CNN，LSTM等有一個快速的方法來找到keras模型中的參數數量？

Answer 1

模型和層具有用於此目的的特殊方法：

model.count_params() 

此外，爲了獲得每一層的尺寸和參數的簡短摘要，你可能會發現有用以下方法

model.summary() 

Answer 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])