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)