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

Less verbose training

Browse Source
This commit is contained in:
Dominik Moritz Roth 2021-09-26 23:15:50 +02:00
parent ade61980b4
commit 9f9d2390b3
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
caliGraph.py
View File

@ -837,8 +837,7 @@ def train(gamma = 1, full=True):
while gamma > 1.0e-06 and delta > 1.0e-06: while gamma > 1.0e-06 and delta > 1.0e-06:
last_mse = mse last_mse = mse
print({'mse': mse, 'w': weights, 'gamma': gamma, 'delta': delta}) print({'mse': mse, 'gamma': gamma, 'delta': delta})
print(gradient)
delta = sum(gradient[g]**2 for g in gradient) delta = sum(gradient[g]**2 for g in gradient)
for wt in weights: for wt in weights:
weights[wt] += gamma*gradient[wt] weights[wt] += gamma*gradient[wt]