From 31fa076b50d0600c6c12c36ad892eaf3735e4e75 Mon Sep 17 00:00:00 2001 From: Vsevolod Stakhov Date: Wed, 3 Jul 2019 14:02:44 +0100 Subject: [Minor] Neural: Fix training with kann --- src/plugins/lua/neural.lua | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'src') 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 -- cgit v1.2.3