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

Bug fix in training

Browse Source
This commit is contained in:
Dominik Moritz Roth 2021-09-27 00:22:35 +02:00
parent b771204b1b
commit 8c4e35bf41
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
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-06 or best_mse > 3: while gamma > 1.0e-06 and delta > 1.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)
@ -852,6 +852,7 @@ def train(gamma = 1, full=True):
else: else:
stagLen += 1 stagLen += 1
if stagLen == 3 or mse > 100: if stagLen == 3 or mse > 100:
stagLen = -2
for wt in weights: for wt in weights:
weights[wt] = random.random() weights[wt] = random.random()
print('Done.') print('Done.')