在喀拉拉邦,有一个使用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的对象。据我了解,Embedding是一个矩阵。那我怎么看那个矩阵呢?

最佳答案

使用model.layers[0].get_weights()[0]Embedding是模型的第一层,要查看的矩阵是该层的第一(也是唯一)权重。

关于python - 查看keras嵌入层的输出,我们在Stack Overflow上找到一个类似的问题:https://stackoverflow.com/questions/46295331/

10-11 15:46