From 9d709ccb69bf65ca4ef09027d21215332bd6b8ab Mon Sep 17 00:00:00 2001
From: M66B <M66B@users.noreply.github.com>
Date: Sun, 3 Jan 2021 11:11:24 +0100
Subject: [PATCH] Optimization

---
 .../main/java/eu/faircode/email/MessageClassifier.java    | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/app/src/main/java/eu/faircode/email/MessageClassifier.java b/app/src/main/java/eu/faircode/email/MessageClassifier.java
index 2c09599107..c6ac5ef93b 100644
--- a/app/src/main/java/eu/faircode/email/MessageClassifier.java
+++ b/app/src/main/java/eu/faircode/email/MessageClassifier.java
@@ -249,15 +249,17 @@ public class MessageClassifier {
 
         JSONArray jwords = new JSONArray();
         for (Long account : classMessages.keySet())
-            for (String word : wordClassFrequency.get(account).keySet())
-                for (String clazz : wordClassFrequency.get(account).get(word).keySet()) {
+            for (String word : wordClassFrequency.get(account).keySet()) {
+                Map<String, Integer> classFrequency = wordClassFrequency.get(account).get(word);
+                for (String clazz : classFrequency.keySet()) {
                     JSONObject jword = new JSONObject();
                     jword.put("account", account);
                     jword.put("word", word);
                     jword.put("class", clazz);
-                    jword.put("frequency", wordClassFrequency.get(account).get(word).get(clazz));
+                    jword.put("frequency", classFrequency.get(clazz));
                     jwords.put(jword);
                 }
+            }
 
         JSONObject jroot = new JSONObject();
         jroot.put("messages", jmessages);