java - 加载自己的数据集时出错。numClasses 和 InputSplit

Question

加载自己的数据集时出现意外异常。执行FeaturizedPreSave中的代码，只需对神经网络要考虑的样本数量进行微小更改，就会产生 Unsupported Operation 异常。

numClasses = 96 

而不是最初的 5. 并拥有DataSet。

错误：

阅读器输出：96 个输出类，但 array.size(1) 为 58（必须等于 1 或 numClasses = 96）。

版本信息：

• Deeplearning4j 1.0.0-alpha
• CUDA 8
• 爪哇 8

痕迹：

• 错误跟踪

• 日志跟踪

Answer 1

从日志看来，测试集中有 58 个示例。所以也许你在某个地方交换了你的轴。

Answer 2

这是我的错！我没有更改新数据集中图像的大小。