From 69c9402d15216fce58494083f4b8a16c44ede31f Mon Sep 17 00:00:00 2001 From: Vsevolod Stakhov Date: Sat, 30 Apr 2016 12:11:18 +0100 Subject: [PATCH] [Feature] Save bayes probability in memory pool var Issue: #613 --- src/libstat/classifiers/bayes.c | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/src/libstat/classifiers/bayes.c b/src/libstat/classifiers/bayes.c index 867fe4dc6..4e30f191e 100644 --- a/src/libstat/classifiers/bayes.c +++ b/src/libstat/classifiers/bayes.c @@ -196,7 +196,7 @@ bayes_classify (struct rspamd_classifier * ctx, GPtrArray *tokens, struct rspamd_task *task) { - double final_prob, h, s; + double final_prob, h, s, *pprob; char *sumbuf; struct rspamd_statfile *st = NULL; struct bayes_task_closure cl; @@ -273,6 +273,9 @@ bayes_classify (struct rspamd_classifier * ctx, } } + pprob = rspamd_mempool_alloc (task->task_pool, sizeof (*pprob)); + rspamd_mempool_set_variable (task->task_pool, "bayes_prob", pprob, NULL); + if (cl.processed_tokens > 0 && fabs (final_prob - 0.5) > 0.05) { sumbuf = rspamd_mempool_alloc (task->task_pool, 32); -- 2.39.5