在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
的对象。在我读到的时候,Embedding
是一个矩阵。那我怎么能看到那个矩阵?
答案 0 :(得分:2)
使用model.layers[0].get_weights()[0]
。 Embedding
是模型的第一层,您想要查看的矩阵是此图层的第一个(也是唯一的)权重。