Tensorflow报错：InvalidArgumentError:indices[401,33] = 77571 is not in [0, 10000)

最新推荐文章于 2023-09-19 22:55:25 发布

原创

最新推荐文章于 2023-09-19 22:55:25 发布 · 1.4w 阅读

5 ·

CC 4.0 BY-SA版权

文章标签：

#Tensorflow #Python

在影评文本分类中遇到Tensorflow报错：indices不在指定范围内。错误源于RNN网络定义的输入长度与实际输入数据长度不匹配。解决方法是调整RNN网络的输入长度以匹配数据。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

在进行影评文本分类时，遇到一个报错，记录下解决过程。

报错信息：

InvalidArgumentError：

indices[401,33] = 77571 is not in [0, 10000)

[[{ {node embedding/embedding_lookup}} = ResourceGather[Tindices=DT_INT32, dtype=DT_FLOAT, validate_indices=true, _device="/job:localhost/replica:0/task:0/device:CPU:0"](embedding/embeddings, embedding/Cast)]]

后在github上看了这边帖子帖子，再分析了一下编写的RNN代码流程：

1、首先RNN网络第一层定义的的输入是10000

vocab_size = 10000

model = tf.keras.Sequential()
model.add(tf.keras.layers.Embedding(vocab_size, 16))
model.add(tf.keras.layers.GlobalAveragePooling1D())
model.add(tf.keras.layers.Dense(16, activation=tf.nn.relu))
model.add(tf.keras.layers.Dense(1, activation=tf.nn.sigmoid))

model.summary()

2、后续进行模型训练的