小编fre*_*cal的帖子

keras 一维卷积输入形状

我正在尝试为 1D 卷积创建模型,但我似乎无法获得正确的输入形状。这是我所拥有的:

#this is actually shape (6826, 9000) but I am shortening it
train_dataset_x = np.array([[0, 1, 5, 1, 10], [0, 2, 4, 1, 3]])
#this is actually shape (6826, 1)
train_dataset_y = np.array([[0], [1]])

model.add(Conv1D(32, 11, padding='valid', activation='relu', strides=1, input_shape=( len(train_dataset_x[0]), train_dataset_x.shape[1]) ))
model.add(Conv1D(32, 3, padding='valid', activation='relu', strides=1) )
model.add(MaxPooling1D())

model.add(Conv1D(64, 3, padding='valid', activation='relu', strides=1) )
model.add(Conv1D(64, 3, padding='valid', activation='relu', strides=1) )
model.add(MaxPooling1D())


model.add(Flatten())
model.add(Dense(256, activation='relu'))
model.add(Dense(1, activation='sigmoid'))
Run Code Online (Sandbox Code Playgroud)

我收到此错误:

ValueError: Error when checking input: expected conv1d_1_input to have …
Run Code Online (Sandbox Code Playgroud)

convolution conv-neural-network keras tensorflow

7
推荐指数
1
解决办法
6869
查看次数