关于Keras嵌入层的快速问题

时间:2018-12-28 22:29:35

标签: keras-layer

我对Keras嵌入层的参数有一个快速的疑问。

嵌入层的input_dimvocab_len,其计算公式为:

vocab_len = len(word_to_index) + 1

为什么我们需要在这里加1?

任何帮助将不胜感激。

谢谢!

1 个答案:

答案 0 :(得分:0)

这是因为嵌入层期望input_dim应该是词汇表的总长度。由于列表的索引为零,因此您需要添加1

来自Keras Documentation

  

input_dim:int>0。词汇量,即最大整数索引   + 1。