pybrain - PyBrain トレーニングを保存して復元するには?

Question

スクリプトを実行するたびに再トレーニングする必要がないように、PyBrain でトレーニング済みのニューラル ネットワークを保存して回復する方法はありますか?

Answer 1

PyBrain のニューラル ネットワークは、python の組み込み pickle/cPickle モジュールを使用するか、PyBrain の 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)

cPickle は C で実装されているため、pickle よりもはるかに高速であることに注意してください。使い方は pickle とほぼ同じなので、代わりに cPickle をインポートして使用してください。

# 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')