Shorter Training

This commit is contained in:
Dominik Moritz Roth 2022-01-22 18:12:57 +01:00
parent efc309c962
commit e5d2ca532c

View File

@ -953,6 +953,7 @@ def train(initGamma, full=True):
goal = 1.0e-5 goal = 1.0e-5
while gamma > goal and delta > goal or best_mse > 15: while gamma > goal and delta > goal or best_mse > 15:
goal *= 1.1
last_mse = mse last_mse = mse
print({'mse': mse, 'gamma': gamma, 'delta': delta}) print({'mse': mse, 'gamma': gamma, 'delta': delta})
delta = sum(gradient[g]**2 for g in gradient) delta = sum(gradient[g]**2 for g in gradient)