Test model: improved output log

2025-12-21 17:45:10 +01:00
parent 4bd5cffd6b
commit b29937c149
1 changed files with 6 additions and 3 deletions
@@ -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}")