如何保存和恢復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') 

Answer 2

NetworkWriter和NetworkReader工作很好。我注意到在通過pickle進行保存和加載時，通過訓練功能不再可以改變網絡。因此，我會推薦使用NetworkWriter-方法。

Answer 3

NetworkWriter是要走的路。按照Jorg的說法，使用Pickle你不能再培訓網絡。

你需要的東西是這樣的：

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

net = buildNetwork(4,6,1) 

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