How to do supervised deepbelief training in PyBrain?

后端 未结 1 555
隐瞒了意图╮
隐瞒了意图╮ 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)
提交回复
热议问题