added workdir parameter

src/rbm/stack.py

@@ -1,3 +1,4 @@
+import os
 from enum import Enum
 from layer import Layer
 
@@ -9,10 +10,12 @@ class StackException(Exception):
 	pass
 
 class Stack:
-	def __init__(self, stack_type: StackType, name: str):
+	def __init__(self, stack_type: StackType, name: str, work_dir: str = '.'):
 		self.stack_type = stack_type
 		self.name = name
+		self.work_dir = work_dir
 		self.layers: list[Layer] = []
+		os.makedirs(self.work_dir, exist_ok=True)
 
 	def num_layers(self):
 		return len(self.layers)
@@ -44,8 +47,10 @@ class Stack:
 
 	def state_save(self):
 		for index, layer in enumerate(self.layers):
-			layer.save(f"{self.name}-{index}-state.npz")
+			filepath = os.path.join(self.work_dir, f"{self.name}-{index}-state.npz")
+			layer.save(filepath)
 
 	def state_load(self):
 		for index, layer in enumerate(self.layers):
-			layer.load(f"{self.name}-{index}-state.npz")
+			filepath = os.path.join(self.work_dir, f"{self.name}-{index}-state.npz")
+			layer.load(filepath)