mirror
/
FairEmail
mirror of https://github.com/M66B/FairEmail.git
1
0
Fork 0
Code Issues Releases Wiki Activity

Limit classifier input

This commit is contained in:
M66B 2021-01-11 14:32:16 +01:00
parent 7eb5dd9e62
commit 443f34f12e
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
app/src/main/java/eu/faircode/email/MessageClassifier.java
View File

@ -54,6 +54,8 @@ public class MessageClassifier {
private static final Map<Long, Map<String, Integer>> classMessages = new HashMap<>();
private static final Map<Long, Map<String, Map<String, Frequency>>> wordClassFrequency = new HashMap<>();
private static final int MAX_WORDS = 1000;
static void classify(EntityMessage message, EntityFolder folder, EntityFolder target, Context context) {
try {
if (!isEnabled(context))
@ -184,6 +186,8 @@ public class MessageClassifier {
for (int end = boundary.next(); end != java.text.BreakIterator.DONE; end = boundary.next()) {
String word = text.substring(start, end);
processWord(account, added, word, state);
if (state.words.size() >= MAX_WORDS)
break;
start = end;
}
} else {
@ -194,6 +198,8 @@ public class MessageClassifier {
for (int end = boundary.next(); end != android.icu.text.BreakIterator.DONE; end = boundary.next()) {
String word = text.substring(start, end);
processWord(account, added, word, state);
if (state.words.size() >= MAX_WORDS)
break;
start = end;
}
}