Неверный аргумент: индексы [207,1] = 1611 отсутствуют в [0, 240) - Tensorflow 2.x (Python)

Question

Я использую модель RNN LSTM для классификации типов личности. Я получаю неожиданную ошибку индексов, когда начинаю тренировать модель. Я пытался использовать некоторые решения, используя трассировки, но нет информации об этой проблеме с использованием TF 2.0. Я оставлю свой Google Colab Если вы хотите посмотреть вверх.

МОДЕЛЬ:

model = keras.Sequential()
model.add(keras.layers.Embedding(input_dim = 240, output_dim = 64)) # The maxlen of the training and validation data is 240.
model.add(keras.layers.Bidirectional(keras.layers.LSTM(64)))
model.add(keras.layers.Dense(64, activation = 'relu'))
model.add(keras.layers.Dense(16, activation = 'softmax'))
model.compile(loss='sparse_categorical_crossentropy', optimizer='adam', metrics=['accuracy'])
model.summary()

Обучение Модель:

fitModel = model.fit(train_data_padded, train_label_seq,
                     epochs = 10,
                     batch_size = 295, # The length of the data is 295
                     validation_data = (validation_padded, validation_label_seq),
                     verbose = 1)

Traceback:

Train on 236 samples, validate on 59 samples
Epoch 1/10
236/236 [==============================] - 0s 71us/sample
---------------------------------------------------------------------------
InvalidArgumentError                      Traceback (most recent call last)
<ipython-input-33-48497b2e653e> in <module>()
      3                                          batch_size = 295, #how many we will load it at once (number of samples per gradient)
      4                                          validation_data = (validation_padded, validation_label_seq), #(x_val, y_val) validation_padded, validation_label_seq
----> 5                      verbose = 1)

11 frames
/usr/local/lib/python3.6/dist-packages/six.py in raise_from(value, from_value)

InvalidArgumentError: 2 root error(s) found.
  (0) Invalid argument:  indices[207,1] = 1611 is not in [0, 240)
     [[node sequential_3/embedding_3/embedding_lookup (defined at <ipython-input-31-bd83004f8334>:5) ]]
  (1) Invalid argument:  indices[207,1] = 1611 is not in [0, 240)
     [[node sequential_3/embedding_3/embedding_lookup (defined at <ipython-input-31-bd83004f8334>:5) ]]
     [[VariableShape/_22]]
0 successful operations.
0 derived errors ignored. [Op:__inference_distributed_function_22239]

Errors may have originated from an input operation.
Input Source operations connected to node sequential_3/embedding_3/embedding_lookup:
 sequential_3/embedding_3/embedding_lookup/20179 (defined at /usr/lib/python3.6/contextlib.py:81)

Input Source operations connected to node sequential_3/embedding_3/embedding_lookup:
 sequential_3/embedding_3/embedding_lookup/20179 (defined at /usr/lib/python3.6/contextlib.py:81)

Function call stack:
distributed_function -> distributed_function