[Fix] Change neural plugin's loss function

Modified the neural network to use ceb_neg loss function.

diff --git a/src/plugins/lua/neural.lua b/src/plugins/lua/neural.lua
index e3518d3bd78dcbdb72b8b6a25b70cfb6d61f7f39..99efe720eb8e839edf006c98d9f6ea4c546c8c21 100644 (file)
--- 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