如何在PyBrain中进行监督深度信念训练

Question

我很难让DeepBeliefTrainer在PyBrain/Python中处理我的数据。由于我找不到任何例子，除了如何在PyBrain中使用无监督的深度学习，我希望有人能给出例子，以显示使用的基本概念。

我尝试使用:

初始化

epochs = 100
layerDims = [768,100,100,1]
net = buildNetwork(*layerDims)
dataset = self.dataset
trainer = DeepBeliefTrainer(net, dataset=dataSet)
trainer.trainEpochs(epochs)

我尝试使用一个有监督的数据集进行回归，但是训练失败了。有人成功地使用深度学习训练器进行监督机器学习吗?你是怎么做到的?

错误:

File "/Library/Python/2.7/site-packages/PyBrain-0.3.1-py2.7.egg/pybrain/structure/networks/rbm.py", line 39, in __init__
self.con = self.net.connections[self.visible][0]
KeyError: None

Answer 1

这是因为你的初始网络:net = buildNetwork(*layerDims)在你的深度信念网络中没有一个以可见层命名的层，即"可见"层。因此，为了在初始网络中找到它的映射，您可以这样做:

net.addInputModule(LinearLayer(input_dim, 'visible'))
[...]
trainer = DeepBeliefTrainer(net, dataset=dataSet)