diff --git a/caliGraph.py b/caliGraph.py index 209d3b8..ebef49b 100755 --- a/caliGraph.py +++ b/caliGraph.py @@ -837,8 +837,7 @@ def train(gamma = 1, full=True): while gamma > 1.0e-06 and delta > 1.0e-06: last_mse = mse - print({'mse': mse, 'w': weights, 'gamma': gamma, 'delta': delta}) - print(gradient) + print({'mse': mse, 'gamma': gamma, 'delta': delta}) delta = sum(gradient[g]**2 for g in gradient) for wt in weights: weights[wt] += gamma*gradient[wt]