diff --git a/src/tests/test_model.py b/src/tests/test_model.py index 1abe717..166d85f 100644 --- a/src/tests/test_model.py +++ b/src/tests/test_model.py @@ -43,8 +43,11 @@ if __name__ == "__main__": # save state model.save() - for inp in batch: - out = model.backward(model.forward(inp)) - print(f"Pattern: {inp} -> {out}") + for index, inp in enumerate(batch): + out = model.backward(model.forward(inp))[0] + 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}")