Keras:嵌入图层+ LSTM:时间维度

时间:2017-08-03 15:02:09

标签: python machine-learning neural-network keras lstm

这可能太愚蠢了......但是......

在Keras的初始Embedding图层之后使用LSTM时(例如Keras LSTM-IMDB tutorial code),Embedding图层如何知道存在时间维度?换句话说,Embedding layer know训练数据集中每个序列的长度如何?嵌入层如何知道我正在训练句子,而不是单个单词?它只是在培训过程中推断出来吗?

1 个答案:

答案 0 :(得分:1)

Embedding图层通常是模型的第一层或第二层。如果它是第一个(通常在您使用Sequential API时) - 那么您需要指定其输入形状,即(seq_len,)(None,)。在它是第二层的情况下(通常在您使用Functional API时),您需要指定第一层是Input层。对于此图层 - 您还需要指定形状。在形状为(None,)的情况下,从输入到模型的一批数据的大小推断出输入形状。