Рассчитать параметры для модельного слоя для Keras MLP - PullRequest
0 голосов
/ 30 марта 2020

Я пытаюсь следовать этому SO сообщению о том, как рассчитываются параметры для каждого слоя, кто-нибудь может дать мне совет?

Вот вывод моих model.summary():

enter image description here

Это модель:

model = Sequential()
model.add(Dense(60, input_dim=44, kernel_initializer='normal', activation='relu'))
model.add(Dense(55, kernel_initializer='normal', activation='relu'))
model.add(Dense(50, kernel_initializer='normal', activation='relu'))
model.add(Dense(45, kernel_initializer='normal', activation='relu'))
model.add(Dense(30, kernel_initializer='normal', activation='relu'))
model.add(Dense(20, kernel_initializer='normal', activation='relu'))
model.add(Dense(1, kernel_initializer='normal'))

1 Ответ

1 голос
/ 30 марта 2020

Для MLPs уравнение имеет вид:

(previous_layer_nodes + 1) * (layer_nodes)

, где +1 означает смещение.

Для входного слоя число узлов предыдущего уровня равно input_dim, поскольку на самом деле вход является неявным слоем .

Итак, в вашем случае:

dense   : (44+1)*60 = 2700
dense_1 : (60+1)*55 = 3355
dense_2 : (55+1)*50 = 2800
dense_3 : (50+1)*45 = 2295
dense_4 : (45+1)*30 = 1380
dense_5 : (30+1)*20 = 620
dense_6 : (20+1)*1  = 21
...