ValueError: 层序的输入 0 与层不兼容:预期 ndim=3,发现 ndim=4。收到的完整形状:[None, 25, 25, 1]

数据挖掘 张量流 卷积
2022-02-23 23:14:35

我正在尝试使用 conv1D 但收到该错误。我的数据集是批处理的,形状为 [None, 25, 25, 1] 我正在使用 input_shape=(25,25) 我无法弄清楚我应该改变什么,这样我才能让它工作。

我的模型:

model = Sequential()
model.add(Conv1D(32, kernel_size=3, activation='relu', input_shape=(25,25))
model.add(Flatten())
model.add(Dense(1, activation='sigmoid'))
1个回答

我通过使用以下方法更改数据集的形状解决了这个问题:

tf.reshape(data, [25, 25])