Machine learning 如何在pybrain中设置自定义hiddenclass函数？

我想用（1，Nh，1，1）训练一个神经网络（一个输入，Nh神经元在第一个隐藏层，1个神经元在第二个隐藏层和1个输出）

在第二个隐藏层中，我想使用一个自定义函数

有什么简单的方法吗？ 我用的是pybrain

---

谢谢

您需要使用自定义的向前和向后逻辑实现从NeuronLayer派生的自己的层。比如：

from pybrain.structure.modules.neuronlayer import NeuronLayer

class CustomLayer(NeuronLayer):
"""Layer implementing the custom function."""

def _forwardImplementation(self, inbuf, outbuf):
    outbuf[:] = custom_func_fwd(inbuf)

def _backwardImplementation(self, outerr, inerr, outbuf, inbuf):
    inerr[:] = custom_func_bkwd(outbuf,outerr)