Refactoring

This commit is contained in:
M66B 2021-01-06 11:39:41 +01:00
parent aff42755b1
commit ae1cd7a592
1 changed files with 7 additions and 7 deletions

View File

@ -170,7 +170,7 @@ public class MessageClassifier {
}
}
private static String classify(long account, String classify, String text, boolean added, Context context) {
private static String classify(long account, String currentClass, String text, boolean added, Context context) {
int maxMatchedWords = 0;
List<String> words = new ArrayList<>();
Map<String, Stat> classStats = new HashMap<>();
@ -208,16 +208,16 @@ public class MessageClassifier {
maxMatchedWords = stat.matchedWords;
}
Integer c = classFrequency.get(classify);
Integer c = classFrequency.get(currentClass);
c = (c == null ? 1 : c + 1);
classFrequency.put(classify, c);
classFrequency.put(currentClass, c);
} else {
Integer c = (classFrequency == null ? null : classFrequency.get(classify));
Integer c = (classFrequency == null ? null : classFrequency.get(currentClass));
if (c != null)
if (c > 0)
classFrequency.put(classify, c - 1);
classFrequency.put(currentClass, c - 1);
else
classFrequency.remove(classify);
classFrequency.remove(currentClass);
}
}
start = end;
@ -272,7 +272,7 @@ public class MessageClassifier {
if (chances.get(0).chance / chances.get(1).chance >= CHANCE_THRESHOLD)
classification = chances.get(0).clazz;
Log.i("Classifier classify=" + classify + " classified=" + classification);
Log.i("Classifier current=" + currentClass + " classified=" + classification);
return classification;
}