如何从嵌入层获取输出

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

您可以得到任何一层,不只是嵌入层的输出,描述在这里:

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.