diff options
| author | Vsevolod Stakhov <vsevolod@highsecure.ru> | 2021-04-29 19:44:40 +0100 |
|---|---|---|
| committer | Vsevolod Stakhov <vsevolod@highsecure.ru> | 2021-04-29 19:44:40 +0100 |
| commit | 6244d64b43baa240d63528849a7a47b3f32eccc3 (patch) | |
| tree | 1df61fec9ded5ac696b9f87d9488bfc9edc0e3b7 /src | |
| parent | 82e588390a7f0dc000e74497cfb84e25dcbfafe5 (diff) | |
| download | rspamd-6244d64b43baa240d63528849a7a47b3f32eccc3.tar.gz rspamd-6244d64b43baa240d63528849a7a47b3f32eccc3.zip | |
[Minor] Neural: Allow to have flat classification if needed
Diffstat (limited to 'src')
| -rw-r--r-- | src/plugins/lua/neural.lua | 12 |
1 files changed, 10 insertions, 2 deletions
diff --git a/src/plugins/lua/neural.lua b/src/plugins/lua/neural.lua index ca11d9e66..2ac8df59f 100644 --- a/src/plugins/lua/neural.lua +++ b/src/plugins/lua/neural.lua @@ -121,7 +121,11 @@ local function ann_scores_filter(task) local result = score if not rule.spam_score_threshold or result >= rule.spam_score_threshold then - task:insert_result(rule.symbol_spam, result, symscore) + if rule.flat_threshold_curve then + task:insert_result(rule.symbol_spam, 1.0, symscore) + else + task:insert_result(rule.symbol_spam, result, symscore) + end else lua_util.debugm(N, task, '%s:%s:%s ann score: %s < %s (spam_score_threshold)', rule.prefix, set.name, set.ann.version, symscore, @@ -131,7 +135,11 @@ local function ann_scores_filter(task) local result = -(score) if not rule.ham_score_threshold or result >= rule.ham_score_threshold then - task:insert_result(rule.symbol_ham, result, symscore) + if rule.flat_threshold_curve then + task:insert_result(rule.symbol_ham, 1.0, symscore) + else + task:insert_result(rule.symbol_ham, result, symscore) + end else lua_util.debugm(N, task, '%s:%s:%s ann score: %s < %s (ham_score_threshold)', rule.prefix, set.name, set.ann.version, result, |