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

Test model: improved output log

Browse Source
This commit is contained in:
jens
2025-12-21 17:45:10 +01:00
parent 4bd5cffd6b
commit b29937c149
1 changed files with 6 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/tests/test_model.py
+6 -3
View File
@@ -43,8 +43,11 @@ if __name__ == "__main__":
# save state # save state
model.save() model.save()
for inp in batch: for index, inp in enumerate(batch):
out = model.backward(model.forward(inp)) out = model.backward(model.forward(inp))[0]
print(f"Pattern: {inp} -> {out}") print(f"- Pattern {index} -------------------------")
print(f"Input : {inp}")
print(f"Output : {(out > 0.9).astype(np.float64)}")
print(f"Error : {np.mean((out - inp)**2):0.3f}")