How to save and recover PyBrain training?

pybrain
solartic picture solartic · May 15, 2011 · Viewed 17.3k times · Source

Is there a way to save and recover a trained Neural Network in PyBrain, so that I don't have to retrain it each time I run the script?

Answer

solartic picture solartic · May 15, 2011

PyBrain's Neural Networks can be saved and loaded using either python's built in pickle/cPickle module, or by using PyBrain's XML NetworkWriter.

# Using pickle

from pybrain.tools.shortcuts import buildNetwork
import pickle

net = buildNetwork(2,4,1)

fileObject = open('filename', 'w')

pickle.dump(net, fileObject)

fileObject.close()

fileObject = open('filename','r')
net = pickle.load(fileObject)

Note cPickle is implemented in C, and therefore should be much faster than pickle. Usage should mostly be the same as pickle, so just import and use cPickle instead.

# Using NetworkWriter

from pybrain.tools.shortcuts import buildNetwork
from pybrain.tools.customxml.networkwriter import NetworkWriter
from pybrain.tools.customxml.networkreader import NetworkReader

net = buildNetwork(2,4,1)

NetworkWriter.writeToFile(net, 'filename.xml')
net = NetworkReader.readFrom('filename.xml')

Related questions

ImportError: No module named scipy

I am using Python 2.7 and trying to get PyBrain to work. But I get this error even though scipy is installed - Traceback (most recent call last): File "<stdin>", line 1, in <module> File "/usr/local/lib/…

Read More
Meaning of an Epoch in Neural Networks Training

while I'm reading in how to build ANN in pybrain, they say: Train the network for some epochs. Usually you would set something like 5 here, trainer.trainEpochs( 1 ) I looked for what is that mean , then I conclude that we use …

Read More
Pybrain time series prediction using LSTM recurrent nets

I have a question in mind which relates to the usage of pybrain to do regression of a time series. I plan to use the LSTM layer in pybrain to train and predict a time series. I found an example …

Read More