added option to save the weights of the AE and the Siamese

Author: AmitaiYacobi

setup.cfg

@@ -1,6 +1,6 @@
 [metadata]
 name = spectralnet
-version = 0.1.0
+version = 0.1.1
 author = Amitai
 description = Spectral Clustering Using Deep Neural Networks
 long_description = file: README.md

src/spectralnet/_trainers/_ae_trainer.py

@@ -20,7 +20,10 @@ def __init__(self, config: dict, device: torch.device):
         self.patience = self.ae_config["patience"]
         self.architecture = self.ae_config["hiddens"]
         self.batch_size = self.ae_config["batch_size"]
+        self.weights_dir = "spectralnet/_trainers/weights"
         self.weights_path = "spectralnet/_trainers/weights/ae_weights.pth"
+        if not os.path.exists(self.weights_dir):
+            os.makedirs(self.weights_dir)
 
     def train(self, X: torch.Tensor) -> AEModel:
         self.X = X.view(X.size(0), -1)