Столкнувшись с некоторыми проблемами при обучении модели LSTM в керасе

Question

Этот код дает мне ошибку

ipython-input-99-b865826eb12b> в data_generator (input_set, img_pretrained, batch_size) 61 62 следующий = np.zeros (VOCABULARY_SIZE) ---> 63next [words_to_token [text_list [i + 1]]] = 1 #one hot 64 65 next_words.append (next)

IndexError: у индекса 3000 нет границ для оси 0 с размером 1000

Получил код от Подпись изображения

batch_size = 128
training_size = get_no_samples(cap_words_cleaned_more, img_features_training)
predictive.compile(loss='categorical_crossentropy', optimizer=RMSprop(), metrics=['accuracy'])

if not DEMO:
    file_name = 'Models/image_caption/checkpoint/weights-improvement-{epoch:02d}-{loss:2.5f}.h5'
    checkpoint = ModelCheckpoint(file_name, monitor='loss', verbose=1, save_best_only=False, mode='min')
    predictive.fit_generator(data_generator(cap_words_cleaned_more, img_features_training,batch_size=batch_size), 
                         steps_per_epoch=training_size/batch_size,
                         nb_epoch=30, 
                         verbose=1,
                        callbacks=[checkpoint])