Merge pull request #27 from florencejt/fix/mcvaetolerance

Fixed bug where mcvae patience for early stopping was called self.pat…
florencejt · May 15, 2024 · 43627de · 43627de
2 parents b7df77f + 93d4a14
commit 43627de
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/fusilli/fusionmodels/tabularfusion/mcvae_model.py b/fusilli/fusionmodels/tabularfusion/mcvae_model.py
@@ -123,8 +123,8 @@ def __init__(self, datamodule, k=None, max_epochs=5000, train_subspace=True):
         self.device = torch.device("cpu")
 
         self.num_latent_dims = 10
-        self.patience = 10
-        self.tolerance = 3
+        self.mcvae_patience = 10
+        self.mcvae_tolerance = 3
         if self.datamodule.layer_mods is not None:
             # if MCVAESubspaceMethod is in the keys
             if "MCVAESubspaceMethod" in self.datamodule.layer_mods.keys():