From 53f87c6325b51fabcfedf04700e0b312f6080e32 Mon Sep 17 00:00:00 2001 From: Vsevolod Stakhov Date: Tue, 22 Nov 2016 12:05:31 +0000 Subject: [PATCH] [Fix] Fix fann_redis when number of scores has been changed --- src/plugins/lua/fann_redis.lua | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) 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) -- 2.47.3