From 4a55a7e4d90854d07c8abf836c2b9149d4e8c6cc Mon Sep 17 00:00:00 2001 From: romanamo Date: Thu, 9 May 2024 12:50:43 +0200 Subject: [PATCH] adjusted training process --- uebungen/aufgabe3/uebung3.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/uebungen/aufgabe3/uebung3.py b/uebungen/aufgabe3/uebung3.py index 3555e45..8535ec8 100644 --- 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)