Mirrors/FairEmail
1
0
Fork 0
mirror of https://github.com/M66B/FairEmail.git synced 2026-03-28 03:46:00 +01:00
Code Issues Packages Projects Releases Wiki Activity

Integrate experimental classifier

Browse Source
This commit is contained in:
M66B
2021-01-02 15:57:24 +01:00
parent c8cb37e02b
commit 1eb8d7fa3f
4 changed files with 26 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -20,8 +20,11 @@ package eu.faircode.email;
*/
import android.content.Context;
import android.content.SharedPreferences;
import android.text.TextUtils;
import androidx.preference.PreferenceManager;
import org.jetbrains.annotations.NotNull;
import java.io.File;
@@ -42,10 +45,11 @@ public class MessageClassifier {
private static final double CHANCE_THRESHOLD = 2.0;
static String classify(EntityMessage message, boolean added, Context context) {
DB db = DB.getInstance(context);
SharedPreferences prefs = PreferenceManager.getDefaultSharedPreferences(context);
if (!prefs.getBoolean("classify", BuildConfig.DEBUG))
return null;
if (!message.content)
throw new IllegalArgumentException("Message without content");
DB db = DB.getInstance(context);
EntityFolder folder = db.folder().getFolder(message.folder);
if (folder == null)
@@ -80,14 +84,15 @@ public class MessageClassifier {
m = (m == null ? 1 : m + 1);
classMessages.put(folder.name, m);
} else {
if (m != null)
if (m != null && m > 0)
classMessages.put(folder.name, m - 1);
}
Log.i("Classifier classify=" + folder.name + " messages=" + classMessages.get(folder.name));
return classified;
}
static String classify(String classify, String text, boolean added) {
private static String classify(String classify, String text, boolean added) {
int maxFrequency = 0;
int maxMatchedWords = 0;
List<String> words = new ArrayList<>();
@@ -107,7 +112,10 @@ public class MessageClassifier {
if (!added) {
Integer c = (classFrequency == null ? null : classFrequency.get(classify));
if (c != null)
classFrequency.put(classify, c - 1);
if (c > 0)
classFrequency.put(classify, c - 1);
else
classFrequency.remove(classify);
continue;
}