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

Dont train if we are already perfect

Browse Source
This commit is contained in:
Dominik Moritz Roth 2021-09-26 23:17:03 +02:00
parent 9f9d2390b3
commit 1303f302d3
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

@ -831,8 +831,8 @@ def train(gamma = 1, full=True):
books = loadBooksFromDB() books = loadBooksFromDB()
bestWeights = copy.copy(weights) bestWeights = copy.copy(weights)
mse, gradient = evaluateFitness(books) mse, gradient = evaluateFitness(books)
delta = sum(gradient[g]**2 for g in gradient)
best_mse = mse best_mse = mse
delta = 1
stagLen = 0 stagLen = 0
while gamma > 1.0e-06 and delta > 1.0e-06: while gamma > 1.0e-06 and delta > 1.0e-06:
@ -854,6 +854,7 @@ def train(gamma = 1, full=True):
if stagLen == 3 or mse > 100: if stagLen == 3 or mse > 100:
for wt in weights: for wt in weights:
weights[wt] = random.random() weights[wt] = random.random()
print('Done.')
def saveWeights(weights): def saveWeights(weights):
with open('neuralWeights.json', 'w') as f: with open('neuralWeights.json', 'w') as f: