如何在PyBrain中进行有监督的深度信仰训练?

时间:2013-03-01 15:32:30

标签: python machine-learning neural-network pybrain dbn

我无法让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)

我尝试使用SupervisedDataset进行回归,但训练失败了。有没有人成功使用deeplearning教练进行有监督的机器学习?你是怎么做到的?

我得到错误:

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

1 个答案:

答案 0 :(得分:4)

这是因为你的初始网络: net = buildNetwork(*layerDims)在您的深层置信网络中没有可见图层名称的图层,该图层是“可见的”。因此,为了找到它在初始网络中的映射,您可以执行以下操作:

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