anon/neural_network_from_scratch
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

more reasonable training settings

Browse Source
This commit is contained in:
anon
2024-06-05 11:59:57 +02:00
parent 68f9a6b1a2
commit 7cd0709894
1 changed files with 3 additions and 2 deletions

5
main.py

@ -242,13 +242,14 @@ def pressure():
# something more bloated such as `NN([7, 8, 8, 1], .5)` # something more bloated such as `NN([7, 8, 8, 1], .5)`
# and significantly better than something longer # and significantly better than something longer
# such as NN([7, 4, 4, 4, 1], .5) # such as NN([7, 4, 4, 4, 1], .5)
# (atleast under reasonable training times)
n = NN([7, 4, 4, 1], .5) n = NN([7, 4, 4, 1], .5)
#n = NN.load('2024-06-05=10:56:28.122959') #n = NN.load('2024-06-05=10:56:28.122959')
print(n) print(n)
n.train(training_data, epochs=5000) n.train(training_data, epochs=10000)
print(n) print(n)
#n.save() n.save()
# This makes our denormalization valid as a matrix operation # This makes our denormalization valid as a matrix operation
vd = np.vectorize(samples['denormalize']) vd = np.vectorize(samples['denormalize'])