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

Question

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

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

有没有简单的方法呢？ 我正在使用pybrain。

谢谢！

Answer 1

您需要使用自定义的前向和后向逻辑实现从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)