Small classifier fix / simplification

pull/207/head
M66B 3 years ago
parent d87a925aa2
commit f66e837883

@ -1385,7 +1385,7 @@ class Core {
// De-classify // De-classify
if (!copy) if (!copy)
for (EntityMessage message : messages) for (EntityMessage message : messages)
MessageClassifier.classify(message, folder, target, context); MessageClassifier.classify(message, folder, false, context);
IMAPFolder itarget = (IMAPFolder) istore.getFolder(target.name); IMAPFolder itarget = (IMAPFolder) istore.getFolder(target.name);
@ -1997,7 +1997,7 @@ class Core {
plain_only, plain_only,
message.preview, message.preview,
parts.getWarnings(message.warning)); parts.getWarnings(message.warning));
MessageClassifier.classify(message, folder, null, context); MessageClassifier.classify(message, folder, true, context);
if (body != null) if (body != null)
EntityLog.log(context, "Operation body size=" + body.length()); EntityLog.log(context, "Operation body size=" + body.length());
@ -4261,7 +4261,7 @@ class Core {
parts.isPlainOnly(download_plain), parts.isPlainOnly(download_plain),
message.preview, message.preview,
parts.getWarnings(message.warning)); parts.getWarnings(message.warning));
MessageClassifier.classify(message, folder, null, context); MessageClassifier.classify(message, folder, true, context);
if (stats != null && body != null) if (stats != null && body != null)
stats.content += body.length(); stats.content += body.length();
@ -4428,7 +4428,7 @@ class Core {
if (process) { if (process) {
EntityContact.received(context, account, folder, message); EntityContact.received(context, account, folder, message);
MessageClassifier.classify(message, folder, null, context); MessageClassifier.classify(message, folder, true, context);
} else } else
Log.d(folder.name + " unchanged uid=" + uid); Log.d(folder.name + " unchanged uid=" + uid);
@ -4746,7 +4746,7 @@ class Core {
parts.isPlainOnly(), parts.isPlainOnly(),
message.preview, message.preview,
parts.getWarnings(message.warning)); parts.getWarnings(message.warning));
MessageClassifier.classify(message, folder, null, context); MessageClassifier.classify(message, folder, true, context);
if (stats != null && body != null) if (stats != null && body != null)
stats.content += body.length(); stats.content += body.length();

@ -56,7 +56,7 @@ public class MessageClassifier {
private static final int MAX_WORDS = 1000; private static final int MAX_WORDS = 1000;
static synchronized void classify(EntityMessage message, EntityFolder folder, EntityFolder target, Context context) { static synchronized void classify(EntityMessage message, EntityFolder folder, boolean added, Context context) {
try { try {
if (!isEnabled(context)) if (!isEnabled(context))
return; return;
@ -64,9 +64,6 @@ public class MessageClassifier {
if (!folder.auto_classify_source) if (!folder.auto_classify_source)
return; return;
if (target != null && !target.auto_classify_source)
return;
long start = new Date().getTime(); long start = new Date().getTime();
// Build text to classify // Build text to classify
@ -86,12 +83,13 @@ public class MessageClassifier {
wordClassFrequency.put(folder.account, new HashMap<>()); wordClassFrequency.put(folder.account, new HashMap<>());
// Classify texts // Classify texts
String classified = classify(message, folder.name, texts, target == null, context); String classified = classify(message, folder.name, texts, added, context);
long elapsed = new Date().getTime() - start; long elapsed = new Date().getTime() - start;
EntityLog.log(context, EntityLog.Type.Classification, message, EntityLog.log(context, EntityLog.Type.Classification, message,
"Classifier" + "Classifier" +
" folder=" + folder.name + " folder=" + folder.name +
" added=" + added +
" message=" + message.id + " message=" + message.id +
"@" + new Date(message.received) + "@" + new Date(message.received) +
":" + message.subject + ":" + message.subject +

Loading…
Cancel
Save