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 /lualib | |
| parent | 82e588390a7f0dc000e74497cfb84e25dcbfafe5 (diff) | |
| download | rspamd-6244d64b43baa240d63528849a7a47b3f32eccc3.tar.gz rspamd-6244d64b43baa240d63528849a7a47b3f32eccc3.zip | |
[Minor] Neural: Allow to have flat classification if needed
Diffstat (limited to 'lualib')
| -rw-r--r-- | lualib/plugins/neural.lua | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/lualib/plugins/neural.lua b/lualib/plugins/neural.lua index f0d5cf582..557133591 100644 --- a/lualib/plugins/neural.lua +++ b/lualib/plugins/neural.lua @@ -57,6 +57,7 @@ local default_options = { -- Check ROC curve and AUC in the ML literature spam_score_threshold = nil, -- neural score threshold for spam (must be 0..1 or nil to disable) ham_score_threshold = nil, -- neural score threshold for ham (must be 0..1 or nil to disable) + flat_threshold_curve = false, -- use binary classification 0/1 when threshold is reached symbol_spam = 'NEURAL_SPAM', symbol_ham = 'NEURAL_HAM', max_inputs = nil, -- when PCA is used |