- use training data if test data not found

2025-12-17 19:33:14 +01:00
parent b4cd05d803
commit 6c02e3c229
1 changed files with 6 additions and 2 deletions
@@ -75,7 +75,10 @@ def main(prj_name: str = "test"):

 	# Load train data
 	training_data = read_armadillo(os.path.join(prj_root, f"{prj_name}.training.dat"))
-	test_data = read_armadillo(os.path.join(prj_root, f"{prj_name}.test.dat"))
+	try:
+		test_data = read_armadillo(os.path.join(prj_root, f"{prj_name}.test.dat"))
+	except FileNotFoundError:
+		test_data = training_data

 	# Prepare status listener
 	my_status = MyStatus(stack, test_data)
@@ -87,7 +90,8 @@ def main(prj_name: str = "test"):
 	stack.state_save()

 if __name__ == "__main__":
-	main("norb_small_16h_v2")
+#	main("norb_small_16h_v2")
+	main("mnist_2")
 	cv.destroyAllWindows()

 	print("Test: [passed]")