Получение значений тензора слоя после однократного вызова model.predict - PullRequest
0 голосов
/ 03 марта 2020

Как получить тензор выходных активаций одного из слоев в нейронной сети после запуска следующего кода.

  x= model.predict(frame)

1 Ответ

0 голосов
/ 03 марта 2020

Вот краткий пример того, как вы могли бы сделать это, используя Функциональный API Keras :

inputs = Input(shape=(10,))
x = Dense(5)(inputs)
latent_outputs = Dense(2)(x)
x = Dense(5)(latent_outputs)
outputs = Dense(10)(x)

encoder = Model(inputs, latent_outputs)
autoencoder = Model(inputs, outputs)

autoencoder.compile(optimizer='adam', loss='mse')
autoencoder.fit(np.zeros((10, 10)), np.zeros((10, 10)))

latent_output = encoder.predict(np.zeros((1, 10)))
print(latent_output)
...