jens/pyRBM
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

xor: load and save weights

Browse Source
This commit is contained in:
jens
2025-12-16 18:40:51 +01:00
parent 5404f2bce0
commit 6a3c59999e
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/rbm/layer.py
+2
View File
@@ -120,10 +120,12 @@ def xor():
status = Status() status = Status()
layer = RbmLayer("Layer_0", 3, 16, params) layer = RbmLayer("Layer_0", 3, 16, params)
layer.init(0.01) layer.init(0.01)
layer.load()
# Train # Train
training_batch = np.array([[0,1,1], [0,0,0], [1,1,0], [1,0,1]], dtype=np.float64) 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.train(training_batch, cd_jens, status)
layer.save()
# Test # Test
test_batch = np.array([[0,0,0], [0,1,0], [1,0,0], [1,1,0]], dtype=np.float64) test_batch = np.array([[0,0,0], [0,1,0], [1,0,0], [1,1,0]], dtype=np.float64)