classifier: add workaround for division by zero

2015-06-17 19:26:50 +03:00 · 2015-06-17 19:26:50 +03:00 · 4da0cb323f
parent e0ae194ac9
commit 4da0cb323f
1 changed files with 5 additions and 1 deletions
--- a/plugins/af_sort_bayes/lib/class.naivebayesian.php
+++ b/plugins/af_sort_bayes/lib/class.naivebayesian.php
@ -80,7 +80,11 @@
 				$scores[$category] = $data['probability'];
 				// small probability for a word not in the category
 				// maybe putting 1.0 as a 'no effect' word can also be good
-				$small_proba = 1.0 / ($data['word_count'] * 2);
+
+				if ($data['word_count'] > 0)
+					$small_proba = 1.0 / ($data['word_count'] * 2);
+				else
+					$small_proba = 0;

 				reset($tokens);