adjusted training process

2024-05-09 12:50:43 +02:00 · 2024-05-09 12:50:43 +02:00 · 4a55a7e4d9
parent e6e35776b8
commit 4a55a7e4d9
1 changed files with 2 additions and 2 deletions
--- a/uebungen/aufgabe3/uebung3.py
+++ b/uebungen/aufgabe3/uebung3.py
@ -78,9 +78,9 @@ class RBM:
        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
    number = mnist[i] / 255
    rbm.train(number)