CNN (Conv2d)无法将大小为12402的数组重塑为形状(318,39,39,1)

Question

我有一个数据集，我在上面训练CNN模型。我的数据集包含398个样本和330个特征，我使用ExtraTreeclassifier()将特征减少到39个。我用Conv1d创建了一个模型，这很好，但我想用Conv2d创建一个CNN。

如何为Conv2D重塑我的x_train和x_test？

x_train = x_train.reshape(x_train.shape[0],x_train.shape[1],x_train.shape[1],1)
x_test = x_test.reshape(x_test.shape[0],x_test.shape[1],x_test.shape[1],1)

结果如下：

ValueError: cannot reshape array of size 12402 into shape (318,39,39,1)

Answer 1

这里，输入节点的数量不等于输出节点的数量。

什么意思？

12402是您的实际图像节点，而您重塑的节点是(318*39*39*1)==483678，它不等于12402

仅当节点数量相等时，才允许进行整形。你必须以这样一种方式进行重塑，即所有dim的乘积等于12402

看看类似的答案here。