commit 2c873e1: [Fix] Change neural plugin's loss function

Pragadeesh C pchandiran at mimecast.com
Mon Aug 24 13:14:07 UTC 2020


Author: Pragadeesh C
Date: 2020-08-19 02:47:49 -0400
URL: https://github.com/rspamd/rspamd/commit/2c873e1d61925293b471f8b4cdb9d6b7d86fac8c (refs/pull/3478/head)

[Fix] Change neural plugin's loss function
Modified the neural network to use ceb_neg loss function.

---
 src/plugins/lua/neural.lua | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/plugins/lua/neural.lua b/src/plugins/lua/neural.lua
index e3518d3bd..99efe720e 100644
--- a/src/plugins/lua/neural.lua
+++ b/src/plugins/lua/neural.lua
@@ -342,8 +342,8 @@ local function create_ann(n, nlayers)
   local nhidden = math.floor((n + 1) / 2)
   local t = rspamd_kann.layer.input(n)
   t = rspamd_kann.transform.relu(t)
-  t = rspamd_kann.transform.tanh(rspamd_kann.layer.dense(t, nhidden));
-  t = rspamd_kann.layer.cost(t, 1, rspamd_kann.cost.mse)
+  t = rspamd_kann.layer.dense(t, nhidden);
+  t = rspamd_kann.layer.cost(t, 1, rspamd_kann.cost.ceb_neg)
   return rspamd_kann.new.kann(t)
 end
 


More information about the Commits mailing list