From 6ccf708e95d5c9d3a0ddcfe74c69da4afd63b7f5 Mon Sep 17 00:00:00 2001
From: M66B <M66B@users.noreply.github.com>
Date: Wed, 6 Jan 2021 17:52:08 +0100
Subject: [PATCH] Require minimum words for each class

---
 app/src/main/java/eu/faircode/email/MessageClassifier.java | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

diff --git a/app/src/main/java/eu/faircode/email/MessageClassifier.java b/app/src/main/java/eu/faircode/email/MessageClassifier.java
index b179a6a6ac..1c084d0645 100644
--- a/app/src/main/java/eu/faircode/email/MessageClassifier.java
+++ b/app/src/main/java/eu/faircode/email/MessageClassifier.java
@@ -250,14 +250,13 @@ public class MessageClassifier {
             EntityLog.log(context, "Classifier " + c +
                     " frequency=" + stat.totalFrequency + "/" + messages +
                     " matched=" + stat.matchedWords + "/" + maxMatchedWords);
-            chances.add(c);
+            if (stat.matchedWords >= MIN_MATCHED_WORDS)
+                chances.add(c);
         }
 
         if (BuildConfig.DEBUG)
             Log.i("Classifier words=" + TextUtils.join(", ", words));
 
-        if (maxMatchedWords < MIN_MATCHED_WORDS)
-            return null;
         if (chances.size() <= 1)
             return null;