dodox/CaliGraph
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Faster termination when training

Browse Source
This commit is contained in:
Dominik Moritz Roth 2021-09-30 00:08:53 +02:00
parent 1bd0597291
commit ca7b2fd87c
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
caliGraph.py
View File

@ -835,7 +835,7 @@ def train(gamma = 1, full=True):
best_mse = mse best_mse = mse
stagLen = 0 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 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)