Ошибка формы ввода при использовании Keras Functional API с fit_generator - PullRequest
Новая
       6

Ошибка формы ввода при использовании Keras Functional API с fit_generator

0 голосов
/ 08 мая 2018

Я построил модель с использованием Keras Functional API, и она работала корректно при вызове fit на наборе поездов.Теперь я решил изменить модель, чтобы использовать мой генератор 

def data_generator():
    while 1:
        for i in range(len(sequences1)):
            yield ([sequences1[i], sequences2[i]], trainLabels[i])

, и вот пример данных из моего набора данных 

sample = next(data_generator())
print(sample)
print(sample[0][0].shape)
# output:
# ([array([ 0,  0,  0, ..., 10, 14, 16], dtype=int32), array([ 0,  0,  0, ..., 19,  1,  4], dtype=int32)], 1)
# (34350,)

, а вот краткое изложение модели (только первые две части)

__________________________________________________________________________________________________
Layer (type)                    Output Shape         Param #     Connected to                     
==================================================================================================
input_1 (InputLayer)            (None, 34350)        0                                            
__________________________________________________________________________________________________
input_2 (InputLayer)            (None, 34350)        0

но когда я пытаюсь подогнать свою модель, используя этот код 

model.fit_generator(data_generator(), epochs=15, steps_per_epoch=64)

я получаю эту ошибку 

ValueError: Error when checking input: expected input_1 to have shape (34350,) but got array with shape (1,)

Как я могу исправитьэто?

1 Ответ

0 голосов
/ 09 мая 2018

Проблема заключается в том, что генератор должен генерировать данные от партии к партии . Другими словами, sample[0][0].shape должно быть (BATCH_SIZE, 34350), и то же самое относится ко второй последовательности и меткам.

...