python - 如何从嵌入层获取输出

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。