From: Vsevolod Stakhov <vsevolod@highsecure.ru>
Date: Tue, 22 Nov 2016 12:05:31 +0000 (+0000)
Subject: [Fix] Fix fann_redis when number of scores has been changed
X-Git-Tag: 1.4.1~111
X-Git-Url: http://git.ipfire.org/gitweb.cgi?a=commitdiff_plain;h=53f87c6325b51fabcfedf04700e0b312f6080e32;p=thirdparty%2Frspamd.git

[Fix] Fix fann_redis when number of scores has been changed
---

diff --git a/src/plugins/lua/fann_redis.lua b/src/plugins/lua/fann_redis.lua
index fffb5dcd96..1cdda8bef7 100644
--- a/src/plugins/lua/fann_redis.lua
+++ b/src/plugins/lua/fann_redis.lua
@@ -283,7 +283,12 @@ local function create_train_fann(n, id)
   end
 
   if fanns[id].fann then
-    if fanns[id].version % max_usages == 0 then
+    if n ~= fanns[id].fann:get_inputs() then
+      rspamd_logger.infox(rspamd_config, 'recreate ANN %s as it has a wrong number of inputs, version %s', id,
+        fanns[id].version)
+      fanns[id].fann_train = rspamd_fann.create(nlayers, n, n / 2, n / 4, 1)
+      fanns[id].fann = nil
+    elseif fanns[id].version % max_usages == 0 then
       -- Forget last fann
       rspamd_logger.infox(rspamd_config, 'recreate ANN %s, version %s', id,
         fanns[id].version)