我有一个train_dataset 4d数组，包含7000张RGB图像（7000,299,299,3）。现在我需要将这个数据集数组输入到sklearn.linear_model.LogisticRegression 中，为此它需要是二维的。

我尝试了::–

np.reshape(train_dataset[:],(7000, train_dataset.shape[1]*train_dataset.shape[2]))

结果是::–

ValueError: cannot reshape array of size 1877152797 into shape (7000,89401)

你能帮我调整这个train_dataset数组的形状吗？我希望保持train_dataset数组中有7000行。

回答：

看起来你只有6999张图像，真遗憾。最简单的方法是让numpy帮你计算缺失的形状

np.reshape(train_dataset, (6999, -1))

或者更稳健的方法

np.reshape(train_dataset, (train_dataset.shape[0], -1))