出错信息
tensorflow.python.framework.errors_impl.InvalidArgumentError: indices[4,0] = 5 is not in [0, 5)
- indices[4,0] = 5 is not in [0, 5)说明embedding的最大索引值为4,这里是5,所以出错了。
- 查看构建Embedding层的代码发现input_dim为4,改为5后即可。
# 出错的代码
embedding = Embedding(input_dim=4, output_dim=8)
# 修改后的正确代码
embedding = Embedding(input_dim=5, output_dim=8)