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