我正在尝试使用keras中的句子嵌入进行简单的文档分类。
我知道如何将单词向量提供给网络,但我在使用句子嵌入时遇到了问题。在我的例子中,我有一个简单的句子表示(沿着轴添加单词向量,例如np.sum(sequences, axis=0)
)。
我的问题是,我应该在下面的代码中替换Embedding
图层来替代句子嵌入?
model = Sequential()
model.add(Embedding(len(embedding_weights), len(embedding_weights[0]), weights=[embedding_weights], mask_zero=True,
input_length=MAX_SEQUENCE_LENGTH, trainable=True))
model.add(LSTM(LSTM_SIZE, activation='relu'))
我已尝试Embedding
图层(未设置权重)和Input
图层但两者都出错。