ValueError: установка элемента массива с последовательностью. Нейронная сеть

я смотрю на какой-то код, пытаясь понять, как он работает, это для нейронной сети, которая использует массивы numpy, я изменил набор данных с меньшим количеством изображений, но теперь я получаю эту ошибку в массиве numpy: " ValueError: установка массива элемент с последовательностью. " Кто-нибудь может сказать мне, в чем проблема, или вам нужен весь код?

train = train_data[:-1]
test = train_data[-191:]

X = np.array([i[0] for i in train]).reshape(-1,IMG_SIZE,IMG_SIZE,1)
Y = [i[1] for i in train]

test_x = np.array([i[0] for i in test]).reshape(-1,IMG_SIZE,IMG_SIZE,1)
test_y = [i[1] for i in test]

model.fit({'input': X}, {'targets': Y}, n_epoch=3, validation_set=({'input': test_x}, {'targets': test_y}),
    snapshot_step=500, show_metric=True, run_id=MODEL_NAME)

person HelloStack    schedule 24.06.2017    source источник
comment
Какая линия? Эта ошибка часто является результатом попытки создать массив из объектов или массивов разного размера. Он пытается создать обычный многомерный массив и не может. Посмотрите на dtype и форму всех объектов, используемых в вызове задачи np.array.   -  person hpaulj    schedule 25.06.2017
comment
На model.fit я получаю эту ошибку. Проблема в размерах массивов, но не знаю, как проверить эти размеры и как это исправить, трудно понять, как это работает.   -  person HelloStack    schedule 25.06.2017
comment
Да, теперь я вижу, массивы сильно отличаются из-за разных наборов данных. Я думаю, что понял это   -  person HelloStack    schedule 25.06.2017