diff options
| author | Vsevolod Stakhov <vsevolod@highsecure.ru> | 2016-01-06 14:24:07 +0000 |
|---|---|---|
| committer | Vsevolod Stakhov <vsevolod@highsecure.ru> | 2016-01-06 14:24:07 +0000 |
| commit | df9ada40a53a804d2d90d9dfddc149a68c141a15 (patch) | |
| tree | ecf56c7cf18b8a95b2fc6a38933ea7dd142c432b /src/libstat/classifiers/bayes.c | |
| parent | 1622570f58b5f5b184f97cd75a52a98cc0b1721a (diff) | |
| download | rspamd-df9ada40a53a804d2d90d9dfddc149a68c141a15.tar.gz rspamd-df9ada40a53a804d2d90d9dfddc149a68c141a15.zip | |
Add learning implementation.
Diffstat (limited to 'src/libstat/classifiers/bayes.c')
| -rw-r--r-- | src/libstat/classifiers/bayes.c | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/src/libstat/classifiers/bayes.c b/src/libstat/classifiers/bayes.c index 0915933f1..b08c70380 100644 --- a/src/libstat/classifiers/bayes.c +++ b/src/libstat/classifiers/bayes.c @@ -303,6 +303,7 @@ bayes_learn_spam (struct rspamd_classifier * ctx, GPtrArray *tokens, struct rspamd_task *task, gboolean is_spam, + gboolean unlearn, GError **err) { guint i, j; @@ -325,7 +326,7 @@ bayes_learn_spam (struct rspamd_classifier * ctx, if (st->stcf->is_spam) { tok->values[id]++; } - else if (tok->values[id] > 0) { + else if (tok->values[id] > 0 && unlearn) { /* Unlearning */ tok->values[id]--; } @@ -334,7 +335,7 @@ bayes_learn_spam (struct rspamd_classifier * ctx, if (!st->stcf->is_spam) { tok->values[id]++; } - else if (tok->values[id] > 0) { + else if (tok->values[id] > 0 && unlearn) { /* Unlearning */ tok->values[id]--; } |