From 6a3c59999e2f7509b8f790f0a6b9dc4c919ac901 Mon Sep 17 00:00:00 2001 From: Jens Ahrensfeld Date: Tue, 16 Dec 2025 18:40:51 +0100 Subject: [PATCH] xor: load and save weights --- src/rbm/layer.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/src/rbm/layer.py b/src/rbm/layer.py index bc188f5..9afee03 100644 --- a/src/rbm/layer.py +++ b/src/rbm/layer.py @@ -120,10 +120,12 @@ def xor(): status = Status() layer = RbmLayer("Layer_0", 3, 16, params) layer.init(0.01) + layer.load() # Train training_batch = np.array([[0,1,1], [0,0,0], [1,1,0], [1,0,1]], dtype=np.float64) layer.train(training_batch, cd_jens, status) + layer.save() # Test test_batch = np.array([[0,0,0], [0,1,0], [1,0,0], [1,1,0]], dtype=np.float64)