如何从嵌入层获取输出

Question

from keras.models import Sequential
from keras.layers.embeddings import Embedding
from theano import function

model = Sequential()
model.add(Embedding(max_features, 128, input_length = maxlen))

我想从嵌入层获取输出。我在keras中阅读了源码，但没有找到任何合适的函数或属性。有人可以帮我这个吗？

Answer 1

您可以获取任何图层的输出，而不仅仅是嵌入图层，如here所述：

from keras import backend as K
get_3rd_layer_output = K.function([model.layers[0].input],
                                  [model.layers[3].output])
layer_output = get_3rd_layer_output([X])[0]

在您的情况下，您需要model.layers[0].output而不是model.layers[3].output。