From ca7b2fd87c0238c73f96230bb8f7419ab1e6fb3c Mon Sep 17 00:00:00 2001
From: Dominik Roth <dominik.roth.dev@gmail.com>
Date: Thu, 30 Sep 2021 00:08:53 +0200
Subject: [PATCH] Faster termination when training

---
 caliGraph.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/caliGraph.py b/caliGraph.py
index 6c3f5cf..90b243a 100755
--- a/caliGraph.py
+++ b/caliGraph.py
@@ -835,7 +835,7 @@ def train(gamma = 1, full=True):
     best_mse = mse
     stagLen = 0
 
-    while gamma > 1.0e-06 and delta > 1.0e-05 or best_mse > 3:
+    while gamma > 3.0e-06 and delta > 3.0e-05 or best_mse > 3:
         last_mse = mse
         print({'mse': mse, 'gamma': gamma, 'delta': delta})
         delta = sum(gradient[g]**2 for g in gradient)