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

Classifier special case: pick first best

This commit is contained in:
M66B 2022-04-29 09:17:44 +02:00
parent bda7ebc8ad
commit 94250953c0
1 changed files with 16 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -263,9 +263,6 @@ public class MessageClassifier {
if (BuildConfig.DEBUG)
Log.i("Classifier words=" + state.words.size() + " " + TextUtils.join(", ", state.words));
if (chances.size() <= 1)
return null;
// Sort classes by chance
Collections.sort(chances, new Comparator<Chance>() {
@Override
@ -278,6 +275,22 @@ public class MessageClassifier {
double class_min_chance = prefs.getInt("class_min_probability", 15) / 100.0;
double class_min_difference = prefs.getInt("class_min_difference", 50) / 100.0;
// Special case: pick first best target class
if (class_min_difference == 0) {
for (Chance chance : chances)
if (chance.chance > class_min_chance) {
EntityFolder target = db.folder().getFolderByName(message.account, chance.clazz);
if (target != null && target.auto_classify_target) {
Log.i("Classifier current=" + currentClass + " classified=" + chance.clazz);
return chance.clazz;
}
}
return null;
}
if (chances.size() <= 1)
return null;
// Select best class
String classification = null;
double c0 = chances.get(0).chance;