Classifier consider

This commit is contained in:
M66B 2021-01-07 09:35:58 +01:00
parent f7a58d9281
commit 7cf2b6aaec
1 changed files with 2 additions and 5 deletions

View File

@ -212,7 +212,7 @@ public class MessageClassifier {
if (!added) if (!added)
return null; return null;
if (state.maxMatchedWords == 0) if (state.maxMatchedWords < MIN_MATCHED_WORDS)
return null; return null;
DB db = DB.getInstance(context); DB db = DB.getInstance(context);
@ -231,16 +231,13 @@ public class MessageClassifier {
} }
Stat stat = state.classStats.get(clazz); Stat stat = state.classStats.get(clazz);
boolean consider = (stat.matchedWords >= MIN_MATCHED_WORDS);
double chance = stat.totalFrequency / messages / state.maxMatchedWords; double chance = stat.totalFrequency / messages / state.maxMatchedWords;
Chance c = new Chance(clazz, chance); Chance c = new Chance(clazz, chance);
EntityLog.log(context, "Classifier " + c + EntityLog.log(context, "Classifier " + c +
" frequency=" + stat.totalFrequency + "/" + messages + " frequency=" + stat.totalFrequency + "/" + messages +
" matched=" + stat.matchedWords + "/" + state.maxMatchedWords + " matched=" + stat.matchedWords + "/" + state.maxMatchedWords +
" consider=" + consider +
" words=" + TextUtils.join(", ", stat.words)); " words=" + TextUtils.join(", ", stat.words));
if (consider) chances.add(c);
chances.add(c);
} }
if (BuildConfig.DEBUG) if (BuildConfig.DEBUG)