From ec6586769fb1faa4a2b3854ffdf69622a9e9fd91 Mon Sep 17 00:00:00 2001
From: Felix Martin <mail@felixm.de>
Date: Fri, 8 May 2026 13:01:09 -0400
Subject: [PATCH] Fix training regression

---
 src/toldg/predict.py | 2 +-
 src/toldg/process.py | 1 +
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/toldg/predict.py b/src/toldg/predict.py
index bf55ad4..1865b27 100644
--- a/src/toldg/predict.py
+++ b/src/toldg/predict.py
@@ -239,7 +239,7 @@ class TransactionClassifier:
             output_dim=len(self.categories),
             learning_rate=0.05,
             reg_lambda=0.01,
-            max_iterations=3000,
+            max_iterations=6000,
         )
 
         self.model.train(x_vectors, label_indices, verbose=verbose)
diff --git a/src/toldg/process.py b/src/toldg/process.py
index 6d98115..33bae46 100644
--- a/src/toldg/process.py
+++ b/src/toldg/process.py
@@ -76,6 +76,7 @@ def apply_mappings(transactions: list[Transaction], mappings: dict[str, Mapping]
             assert mapping.count > 0, f"{mapping} used by {t} but count is not greater than '0'."
             mapping.count -= 1
             t.mapping = mapping
+            t.account2 = mapping.account2
         else:
             unmapped_count += 1
     if unmapped_count > 0: