python - 查看keras嵌入层的输出

Question

在 keras 中，有一个使用 IMDB 数据集进行情感分类的示例。代码如下所示

top_words = 5000
(X_train, y_train), (X_test, y_test) = imdb.load_data(nb_words=top_words)
max_review_length = 500
X_train = sequence.pad_sequences(X_train, maxlen=max_review_length)
X_test = sequence.pad_sequences(X_test, maxlen=max_review_length)
embedding_vecor_length = 32
model = Sequential()
model.add(Embedding(top_words, embedding_vecor_length, input_length=max_review_length))
model.add(LSTM(100))
......

我想查看此行Embedding(top_words, embedding_vecor_length, input_length=max_review_length)的输出。当我尝试在终端中调试或运行该行时，我看到它返回类 Embedding 的对象。据我所知，嵌入是一个矩阵。那么我怎样才能看到这个矩阵呢？

Answer 1

使用model.layers[0].get_weights()[0]。 嵌入是模型的第一层，您想要看到的矩阵是该层的第一个(也是唯一的)权重。