numpy 这里应该做些什么才能在重新整形数组时解决这个问题？

rbl8hiat 于 2023-02-16 发布在其他

关注(0)|答案(1)|浏览(115)

X_test = test_df.drop('Label',axis = 1).values.reshape(test_df.shape[0], 28, 28, 1)

ValueError：无法将大小为5630020的数组整形为（7172，28，28，1）的形状

print(test_df.shape) >> (7172, 786)

我试着改变numpy数组的形状，但它似乎小于2。

1条答案

这个形状看起来像是来自类似MNIST数据集的数据。如果是这样的话，看起来你正在尝试对这个数据做一些分类，所以你能分享一下你是如何定义你的训练/测试集的吗？在我看来，你的问题可能来自这个。