为什么scikit神经网络会自动重塑我的输入数组?

时间:2015-08-18 15:03:06

标签: python arrays numpy scikit-learn neural-network

当我将训练数据传递给scikit神经网络时,它是我所有24 * 24图像矩阵的NumPy数组。我检查,这是数据的形状:(3237,24,24) - 3237 24 * 24图像。

然而,一旦我把它喂入神经网络,我就会收到这个警告:

  

警告:sknn: - 将输入数组从(3237,24,24)重新整形为(2589,24,24,1)。

程序仍在运行,我只是不明白它为什么要重塑我的阵列。

1 个答案:

答案 0 :(得分:0)

查看源代码,如果初始化时传入的X(输入样本)与创建/拆分MLP后端中的数据集的大小不同,则scikit将重新整形输入数组。这是为了通过训练和验证相同的样本来减少过度拟合。