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

View File

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