如何保存经过训练的神经网络(图像中的数字识别)

时间:2011-11-08 07:38:47

标签: java neural-network

我编写了一个神经网络程序,使用反向传播和前馈来训练图像中的识别数字。它的工作非常好。我只是想知道如何保存这个训练有素的神经网络,因为我不知道要保存什么。我的目标是训练我的神经网络一次,因为它需要很长时间训练(1000次迭代,和许多训练集),然后我想保存它并加载它供以后使用。我希望你理解我的问题。我不能透露我的所有神经代码。但这就是执行代码的样子

NeuralNetwork nn = new NeuralNetwork(numberOfInputs, nuberOfHiddenNodes, numberOfOuputNodes);
pattern less than patternsList.length:
nn.train(pattern);

2 个答案:

答案 0 :(得分:1)

在您的网络经过培训后,您可以使用Java Serialization存储(并稍后加载)NeuralNetwork及其子项的内存内容。

答案 1 :(得分:1)

class NeuralNetwork implements Serializable { ... }

使用ObjectOutputStreamFileOutputStream。调用方法writeObject(...)来编写NeuralNetwork对象。