InvalidArgumentError:indexs [120,0] = 3080不在[0,32)中[[{{node embedding_6 / embedding_lookup}}]]

时间:2019-11-25 12:14:53

标签: python keras nlp lstm

我看到其他人也发表了类似的问题。但是不同之处在于,我运行的是Keras Functional API,而不是顺序模型。

from keras.models import Model
from keras import layers
from keras import Input
text_vocabulary_size = 10000
question_vocabulary_size = 10000
answer_vocabulary_size = 500

text_input = Input(shape=(None,), dtype='int32', name='text')
embedded_text = layers.Embedding(64, text_vocabulary_size)(text_input)
encoded_text = layers.LSTM(32)(embedded_text)
question_input = Input(shape=(None,), dtype='int32', name='question')
embedded_question = layers.Embedding( 32, question_vocabulary_size)(question_input)
encoded_question = layers.LSTM(16)(embedded_question)

concatenated = layers.concatenate([encoded_text, encoded_question],axis=-1) 

## Concatenates the encoded question and encoded text

answer = layers.Dense(answer_vocabulary_size, activation='softmax')(concatenated)
model = Model([text_input, question_input], answer)
model.compile(optimizer='rmsprop', loss='categorical_crossentropy', metrics=['acc'])

将数据输入到多输入模型

import numpy as np
num_samples = 1000
max_length = 100

text = np.random.randint(1, text_vocabulary_size, size=(num_samples, max_length))

question = np.random.randint(1, question_vocabulary_size,  size=(num_samples, max_length))

answers = np.random.randint(0, 1, size=(num_samples, answer_vocabulary_size)) 

使用输入列表进行拟合

model.fit([text, question], answers, epochs=10, batch_size=128)

尝试拟合模型时遇到的错误如下。

InvalidArgumentError: indices[120,0] = 3080 is not in [0, 32)
     [[{{node embedding_6/embedding_lookup}}]]

1 个答案:

答案 0 :(得分:0)

词汇量是Embedding类的第一个参数,您将其设置为第二个。您只需要切换要提供给嵌入实例的参数即可。