How to do supervised deepbelief training in PyBrain?

后端 未结 1 569
隐瞒了意图╮
隐瞒了意图╮ 2021-02-03 11:31

I have trouble getting the DeepBeliefTrainer to work on my data in PyBrain/Python. Since I can\'t find any examples other than unsupervised on how to use the deep learning in Py

1条回答
  •  既然无缘
    2021-02-03 12:15

    It's because your initial network: net = buildNetwork(*layerDims) doesn't have a layer with the name of the visible layer in your deep belief network, which is 'visible'. So, in order to find it mapped in the initial network, you can do something like:

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

    0 讨论(0)
提交回复
热议问题