我对Keras嵌入层的参数有一个快速的疑问。
嵌入层的input_dim
为vocab_len
,其计算公式为:
vocab_len = len(word_to_index) + 1
为什么我们需要在这里加1?
任何帮助将不胜感激。
谢谢!
答案 0 :(得分:0)
这是因为嵌入层期望input_dim应该是词汇表的总长度。由于列表的索引为零,因此您需要添加1
来自Keras Documentation:
input_dim:int>0。词汇量,即最大整数索引 + 1。