Require minimum words for each class

pull/191/head
M66B 4 years ago
parent 206a7dd0f2
commit 6ccf708e95

@ -250,14 +250,13 @@ public class MessageClassifier {
EntityLog.log(context, "Classifier " + c + EntityLog.log(context, "Classifier " + c +
" frequency=" + stat.totalFrequency + "/" + messages + " frequency=" + stat.totalFrequency + "/" + messages +
" matched=" + stat.matchedWords + "/" + maxMatchedWords); " matched=" + stat.matchedWords + "/" + maxMatchedWords);
chances.add(c); if (stat.matchedWords >= MIN_MATCHED_WORDS)
chances.add(c);
} }
if (BuildConfig.DEBUG) if (BuildConfig.DEBUG)
Log.i("Classifier words=" + TextUtils.join(", ", words)); Log.i("Classifier words=" + TextUtils.join(", ", words));
if (maxMatchedWords < MIN_MATCHED_WORDS)
return null;
if (chances.size() <= 1) if (chances.size() <= 1)
return null; return null;

Loading…
Cancel
Save