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

Question

加载自己的数据集时出现意外异常。从FeaturizedPreSave执行代码只需要对神经网络要考虑的样本数量进行微小改动就会产生不支持的操作的例外。

numClasses = 96 

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

错误：

  

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

版本信息：

• Deeplearning4j 1.0.0-alpha
• CUDA 8
• Java 8

跟踪：

• error_trace

• log_trace

Answer 1

从日志中可以看出测试集中有58个示例。所以也许你已经在某处交换了你的轴。

Answer 2

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