PyBrain - 如何进行深度置信网络训练?

18

我在使用Pybrain训练DBN时遇到了一些困难。首先,我尝试了简单的方法:

net = buildNetwork(*layerDims)
我遇到了这个问题:如何在PyBrain中进行监督式深度置信网络训练? ,而建议的解决方案只导致另一个错误:


File "/home/WORK/Canopy_64bit/User/lib/python2.7/site-packages/PyBrain-0.3.1-    py2.7.egg/pybrain/unsupervised/trainers/deepbelief.py", line 62, in <genexpr>
layercons = (self.net.connections[i][0] for i in layers)
IndexError: list index out of range

于是我尝试从零开始定义一个网络!

inp = LinearLayer(3 , 'visible')
hidden0 = SigmoidLayer(2 , 'hidden0')
hidden1= SigmoidLayer(2 , 'hidden1')
output = LinearLayer(2 , 'output')
bias = BiasUnit('bias')
net = Network()
net.addInputModule(inp)
net.addModule(hidden0)
net.addModule(hidden1)
net.addModule(output)
net.addModule(bias)
net.addConnection(FullConnection(inp, hidden0))
net.addConnection(FullConnection(hidden0, hidden1))
net.addConnection(FullConnection(hidden1, output))
net.addConnection(FullConnection(bias, hidden0))
net.addConnection(FullConnection(bias, hidden1))
net.addConnection(FullConnection(bias, output))
net.sortModules()

即使我仍在奔跑:

trainer = deepbelief.DeepBeliefTrainer(net1, dataset=ds)
trainer.trainEpochs(epochs)

我看到这个错误:

File "/home/WORK/Canopy_64bit/User/lib/python2.7/site-packages/PyBrain-0.3.1-py2.7.egg/pybrain/structure/connections/connection.py", line 37, in __init__
self.outSliceTo = outmod.indim
AttributeError: 'NoneType' object has no attribute 'indim'

这与相关RBM中的隐藏层有关。

我在这里漏掉了什么吗?


1
您需要制作一个输出模块。因此,addModule(output) 应更改为 addOutputModule(output)。 - jay
3
Pybrain 是一个非常慢的库。我转而使用其他库,如 caffe 或 c 库,它们的速度提高了 25 倍或更多。 - pbu
1
我建议不要使用这个旧的库,而是看看基于Theano的库,比如Blocks - Def_Os
1个回答

1
你可以通过以下代码初始化一个名为net的网络:
net = Network()
net.addInputModule(inp)
net.addModule(hidden0)
net.addModule(hidden1)
net.addModule(output)
net.addModule(bias)
net.addConnection(FullConnection(inp, hidden0))
net.addConnection(FullConnection(hidden0, hidden1))
net.addConnection(FullConnection(hidden1, output))
net.addConnection(FullConnection(bias, hidden0))
net.addConnection(FullConnection(bias, hidden1))
net.addConnection(FullConnection(bias, output))
net.sortModules()

但你将net1作为参数传递:

trainer = deepbelief.DeepBeliefTrainer(net1, dataset=ds)
trainer.trainEpochs(epochs)

这肯定会导致一个错误。

所以

trainer = deepbelief.DeepBeliefTrainer(net, dataset=ds)

应该可以解决你的问题。


网页内容由stack overflow 提供, 点击上面的
可以查看英文原文,
原文链接