我正在尝试使用 Neuroph 库训练神经网络,但出现此错误:
44 [NeurophLearningThread] INFO org.neuroph.core.learning.LearningRule - Learning Started
Exception in thread "AWT-EventQueue-0" java.lang.StackOverflowError
at java.lang.reflect.Method.invoke(Method.java:498)
at java.io.ObjectStreamClass.invokeWriteObject(ObjectStreamClass.java:1028)
at java.io.ObjectOutputStream.writeSerialData(ObjectOutputStream.java:1496)
at java.io.ObjectOutputStream.writeOrdinaryObject(ObjectOutputStream.java:1432)
at java.io.ObjectOutputStream.writeObject0(ObjectOutputStream.java:1178)
at java.io.ObjectOutputStream.defaultWriteFields(ObjectOutputStream.java:1548)
at java.io.ObjectOutputStream.writeSerialData(ObjectOutputStream.java:1509)
at java.io.ObjectOutputStream.writeOrdinaryObject(ObjectOutputStream.java:1432)
at java.io.ObjectOutputStream.writeObject0(ObjectOutputStream.java:1178)
at java.io.ObjectOutputStream.defaultWriteFields(ObjectOutputStream.java:1548)
at java.io.ObjectOutputStream.writeSerialData(ObjectOutputStream.java:1509)
at java.io.ObjectOutputStream.writeOrdinaryObject(ObjectOutputStream.java:1432)
at java.io.ObjectOutputStream.writeObject0(ObjectOutputStream.java:1178)
at java.io.ObjectOutputStream.writeObject(ObjectOutputStream.java:348)
at org.neuroph.util.NeurophArrayList.writeObject(NeurophArrayList.java:710)
at sun.reflect.GeneratedMethodAccessor1.invoke(Unknown Source)
at sun.reflect.DelegatingMethodAccessorImpl.invoke(DelegatingMethodAccessorImpl.java:43)
问题是,我在另一个项目中使用相同的代码训练神经网络,该项目具有相同的导入 jar 并且工作正常。这是我用来创建和训练神经网络的代码:
public void TrainNeuralNetwork(double[] inputs, double[] outputs){
DataSet set = new DataSet(inputs.length,outputs.length);
set.addRow(inputs,outputs);
System.out.println("Training...");
MultiLayerPerceptron loadSnakeNN = new MultiLayerPerceptron(TransferFunctionType.LINEAR, inputs.length, 500, outputs.length);
if( loadSnakeNN.getLearningRule() instanceof MomentumBackpropagation )
((MomentumBackpropagation)loadSnakeNN.getLearningRule()).setBatchMode(true);
MomentumBackpropagation learningRule = (MomentumBackpropagation)loadSnakeNN.getLearningRule();// Set learningRule
learningRule.setMaxError(0.01);
loadSnakeNN.learn(set);
loadSnakeNN.save("SnakeNN.nnet");
System.out.println("Neural Network Trained!");
}
更新
我发现问题出在哪里了。当我使用超过 90 个输入时,我收到上述错误。所以我的新问题是,如何训练这个具有超过 90 个输入的神经网络?
最佳答案
您可以使用 JVM 参数 -Xss
增加程序可用的堆栈大小。
`java -Xss2048m ...`
意味着你得到一个2048Mb
的堆栈。如果您从 IDE 运行此程序,您可以搜索以了解如何为您的特定 IDE 传递 JVM 参数。
我假设您的代码或库中没有错误,否则增加堆栈可能还不够。
关于java - 尝试使用 Neuroph 框架训练神经网络时出现 StackOverFlowError,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/35390429/