2210788
/
gnn
forked from 2211275/gnn
1
0
Fork 0
Code Pull Requests Activity

adjusted training process

main
romanamo 2024-05-09 12:50:43 +02:00
parent e6e35776b8
commit 4a55a7e4d9
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
uebungen/aufgabe3/uebung3.py
View File

@ -78,9 +78,9 @@ class RBM:
return h0, v1 return h0, v1
rbm = RBM(28 ** 2, 100, 0.2, epochs=1) rbm = RBM(28 ** 2, 100, 0.2, epochs=2)
for i in range(100): for i in range(100, 600):
# normalize mnist data and train # normalize mnist data and train
number = mnist[i] / 255 number = mnist[i] / 255
rbm.train(number) rbm.train(number)