神经网络输入形状

时间:2020-09-10 04:28:01

标签: tensorflow deep-learning neural-network nlp

考虑我的语料库是否有5个句子,其中最大句子大小为10个单词。

因此,嵌入矩阵将为5x10。

神经网络的输入层的输入形状将如何变化,或者如何将数据作为神经网络的输入给出

1 个答案:

答案 0 :(得分:0)

您网络中的第一层将是嵌入层。

输入的第一个参数的维度 将是一个数字<= 50(5个句子* 10个单词)。 考虑到数据集中的所有单词可能并非唯一的事实,该值可能会较小。

因此,输入层将是Embedding()层,

input_shape = (number_of_words_in_the_vocabulary, embedding_dimension)

其中嵌入尺寸是您可以选择的超参数。

您可以在这里看看:https://www.tensorflow.org/tutorials/text/word_embeddings