Stack: added state load

src/rbm/rbm.py

@@ -47,12 +47,18 @@ def main():
 	# Create stack from project file
 	stack = StackFactory.from_file("/home/jens/work/repos/Rbm/test.prj")
 
+	# Load state
+	stack.state_load()
+
 	# Load train data
 	batch = read_armadillo("/home/jens/work/repos/Rbm/test.training.dat")
 
 	# Train
 	stack.train(batch)
 
+	# Save state
+	stack.state_save()
+
 if __name__ == "__main__":
 	main()
 	print("Test: [passed]")

src/rbm/stack.py

@@ -45,3 +45,7 @@ class Stack:
 	def state_save(self):
 		for index, layer in enumerate(self.layers):
 			layer.save(f"{self.name}-{index}-state.npz")
+
+	def state_load(self):
+		for index, layer in enumerate(self.layers):
+			layer.load(f"{self.name}-{index}-state.npz")