From 44752a8175c54effd0c901333a7eb4cc71dd6810 Mon Sep 17 00:00:00 2001 From: Determinant Date: Wed, 27 May 2015 12:02:44 +0800 Subject: add implementation for sigmoid layer (not tested) --- layer/sigmoid.lua | 14 ++++++++++++++ 1 file changed, 14 insertions(+) create mode 100644 layer/sigmoid.lua (limited to 'layer/sigmoid.lua') diff --git a/layer/sigmoid.lua b/layer/sigmoid.lua new file mode 100644 index 0000000..d69e9e3 --- /dev/null +++ b/layer/sigmoid.lua @@ -0,0 +1,14 @@ +local SigmoidLayer = nerv.class("nerv.SigmoidLayer", "nerv.Layer") + +function SigmoidLayer:__init(id, global_conf) + self.id = id + self.gconf = global_conf +end + +function SigmoidLayer:propagate(input, output) + output:sigmoid(input) +end + +function SigmoidLayer:back_propagate(next_bp_err, bp_err, input, output) + next_bp_err:sigmoid(bp_err, output) +end -- cgit v1.2.3