CNN（Conv2d）无法将大小为12402的阵列重塑为形状（318,39,39,1）

Question

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

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