From: Vsevolod Stakhov <vsevolod@highsecure.ru>
Date: Wed, 3 Jul 2019 13:02:44 +0000 (+0100)
Subject: [Minor] Neural: Fix training with kann
X-Git-Tag: 2.0~670
X-Git-Url: http://git.ipfire.org/cgi-bin/gitweb.cgi?a=commitdiff_plain;h=31fa076b50d0600c6c12c36ad892eaf3735e4e75;p=thirdparty%2Frspamd.git

[Minor] Neural: Fix training with kann
---

diff --git a/src/plugins/lua/neural.lua b/src/plugins/lua/neural.lua
index 193b076144..5f9b1f9d2a 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