Cách tìm số tham số của mô hình keras?

Question

Đối với mạng nguồn cấp dữ liệu (FFN), bạn có thể dễ dàng tính toán số tham số. Đưa ra một CNN, LSTM vv là có một cách nhanh chóng để tìm số lượng các tham số trong một mô hình keras?

Answer 1

Models và các lớp có phương pháp đặc biệt cho mục đích đó:

model.count_params() 

Ngoài ra, để có được một bản tóm tắt ngắn của mỗi kích thước lớp và các thông số, bạn có thể thấy hữu ích các phương pháp sau đây

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