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

Classifier: check if class exists

This commit is contained in:
M66B 2021-01-05 13:18:17 +01:00
parent fcecc0ed92
commit 05422c6cf9
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -255,12 +255,17 @@ public class MessageClassifier {
if (maxMatchedWords == 0) if (maxMatchedWords == 0)
return null; return null;
DB db = DB.getInstance(context);
List<Chance> chances = new ArrayList<>(); List<Chance> chances = new ArrayList<>();
for (String clazz : classStats.keySet()) { for (String clazz : classStats.keySet()) {
Integer messages = classMessages.get(account).get(clazz); Integer messages = classMessages.get(account).get(clazz);
if (messages == null || messages == 0) if (messages == null || messages == 0)
continue; continue;
EntityFolder folder = db.folder().getFolderByName(account, clazz);
if (folder == null)
continue;
Stat stat = classStats.get(clazz); Stat stat = classStats.get(clazz);
double chance = (double) stat.totalFrequency / messages / maxMatchedWords; double chance = (double) stat.totalFrequency / messages / maxMatchedWords;
Chance c = new Chance(clazz, chance); Chance c = new Chance(clazz, chance);