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

- use training data if test data not found

Browse Source
This commit is contained in:
jens
2025-12-17 19:33:14 +01:00
parent b4cd05d803
commit 6c02e3c229
1 changed files with 6 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/rbm/rbm.py
+6 -2
View File
@@ -75,7 +75,10 @@ def main(prj_name: str = "test"):
# Load train data # Load train data
training_data = read_armadillo(os.path.join(prj_root, f"{prj_name}.training.dat")) 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 # Prepare status listener
my_status = MyStatus(stack, test_data) my_status = MyStatus(stack, test_data)
@@ -87,7 +90,8 @@ def main(prj_name: str = "test"):
stack.state_save() stack.state_save()
if __name__ == "__main__": if __name__ == "__main__":
main("norb_small_16h_v2") # main("norb_small_16h_v2")
main("mnist_2")
cv.destroyAllWindows() cv.destroyAllWindows()
print("Test: [passed]") print("Test: [passed]")