diff options
| author | Vsevolod Stakhov <vsevolod@highsecure.ru> | 2021-04-29 19:41:03 +0100 |
|---|---|---|
| committer | Vsevolod Stakhov <vsevolod@highsecure.ru> | 2021-04-29 19:41:43 +0100 |
| commit | 82e588390a7f0dc000e74497cfb84e25dcbfafe5 (patch) | |
| tree | a21472fe460d3c46c4ac284dad7c3fa12865dda5 /lualib/plugins | |
| parent | b9f9beebe869b5800e80a8dbbc3d49a9c9457062 (diff) | |
| download | rspamd-82e588390a7f0dc000e74497cfb84e25dcbfafe5.tar.gz rspamd-82e588390a7f0dc000e74497cfb84e25dcbfafe5.zip | |
[Feature] Neural: Allow to balance FP/FN for the network
Diffstat (limited to 'lualib/plugins')
| -rw-r--r-- | lualib/plugins/neural.lua | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/lualib/plugins/neural.lua b/lualib/plugins/neural.lua index c35fc0eeb..f0d5cf582 100644 --- a/lualib/plugins/neural.lua +++ b/lualib/plugins/neural.lua @@ -54,6 +54,9 @@ local default_options = { learning_spawned = false, ann_expire = 60 * 60 * 24 * 2, -- 2 days hidden_layer_mult = 1.5, -- number of neurons in the hidden layer + -- 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) symbol_spam = 'NEURAL_SPAM', symbol_ham = 'NEURAL_HAM', max_inputs = nil, -- when PCA is used |