ValueError: вход 0 несовместим со слоем conv2_1: ожидаемый ndim = 4, найденный ndim = 3 - PullRequest
Новая
       25

ValueError: вход 0 несовместим со слоем conv2_1: ожидаемый ndim = 4, найденный ndim = 3

0 голосов
/ 03 декабря 2018

Я новичок в Deep Learning.Я случайно сгенерировал наборы данных со следующей формой (5,4,4).Это что-то вроде

[[[1 2 3 4] [4 5 6 7] [7 8 9 10]]]

Я не знаю, почемуЭто создает проблему, связанную с размерами

Код My Keras приведен ниже 

X_train=np.random.randint(0,100,size=(5,4,4))
Y_train=np.random.rand(5,1)
X_valid=np.random.randint(0,100,size=(2,4,4)
Y_valid=np.random.rand(2,1)

def create_model():
    nb_filters=2
    nb_conv=2
    model=Sequential()
    model.add(Convolution2d(nb_filters,nb_conv,padding='same',input_shape=(4,4)
    model.add(Activation('relu'))
    ****Other layers****
    enter code here
    model.add(Dense(1)
    model.add(Activation('linear')
    model.compile(loss='mean_squared_error', optimizer=Adadelta())
    return model
model=create_model()
model.fit(X_train,Y_train, batch_size=2,nb_epoch=50,verbos=1, validation_data=(X_valid,Y_valid)

1 Ответ

0 голосов
/ 03 декабря 2018

В ваших входных данных отсутствует размер канала (см. документы )

...