如何在R中使用神经网络中的自定义激活函数

时间:2017-12-14 16:30:02

标签: r neural-network

我是R的新手,我正在尝试为回归任务构建一个神经网络。

我使用neuralnet库来构建我的神经网络,我注意到它接受了几个参数。 act.fct是其中之一。

  

act.fct

     

可微函数,用于平滑协变量或神经元的交叉积和重量的结果。此外,逻辑函数和切线双曲线可以使用“logistic”和“tanh”字符串。

默认提供后勤激活功能,但我想使用自定义softplus功能。

我的代码能够正常运行,但我怎么知道是否使用了softplus而不是默认的激活功能?

以下是我的R代码段。

softplus <- function(x) log(1+exp(x))

net <- neuralnet(formula = f, 
             data = as.matrix(train),
             act.fct = softplus,
             stepmax = 1e7,
             hidden = c(10, 8, 6),
             linear.output = T)

1 个答案:

答案 0 :(得分:1)

只需检查对象net

net$act.fct(x)

应该并且将返回与softplus(x)相同的内容。