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



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

然而,一旦我把它输入神经网络,我得到这个警告:

WARNING:sknn: -将输入数组从(3237,24,24)重塑为(2589,24,24,1).

程序还在运行,我只是不明白为什么它在重塑我的数组

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

相关内容

  • 没有找到相关文章

最新更新