Browse Source

[Fix] Change neural plugin's loss function

Modified the neural network to use ceb_neg loss function.
pull/3478/head
Pragadeesh C 5 years ago
parent
commit
2c873e1d61
  1. 4
      src/plugins/lua/neural.lua

4
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

Loading…
Cancel
Save