diff options
| -rw-r--r-- | src/plugins/lua/neural.lua | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/src/plugins/lua/neural.lua b/src/plugins/lua/neural.lua index 193b07614..5f9b1f9d2 100644 --- a/src/plugins/lua/neural.lua +++ b/src/plugins/lua/neural.lua @@ -551,13 +551,13 @@ local function train_ann(rule, _, ev_base, elt, worker) for _,e in ipairs(spam_elts) do if e == e then inputs[#inputs + 1] = e - outputs[#outputs + 1] = 1.0 + outputs[#outputs + 1] = {1.0} end end for _,e in ipairs(ham_elts) do if e == e then inputs[#inputs + 1] = e - outputs[#outputs + 1] = 0.0 + outputs[#outputs + 1] = {0.0} end end |