use consistent naming for configuration options for svae

BIMSBbioinfo · Sep 19, 2023 · bba1a44 · bba1a44
1 parent e8a9343
commit bba1a44
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/examples/configs/hpo_configuration.yaml b/examples/configs/hpo_configuration.yaml
@@ -20,7 +20,7 @@ DirectPred:
     name: epochs
     categories: [50, 100, 150]
 
-SVAE:
+supervised_vae:
   - type: Integer
     name: latent_dim
     min: 16

diff --git a/flexynesis/__main__.py b/flexynesis/__main__.py
@@ -46,7 +46,7 @@ def main():
         config_name = 'DirectPred'
     elif args.model_class == "supervised_vae":
         model_class = flexynesis.supervised_vae
-        config_name = 'SVAE'
+        config_name = 'supervised_vae'
     elif args.model_class == "MultiTripletNetwork":
         model_class = flexynesis.MultiTripletNetwork
         config_name = 'MultiTripletNetwork'

diff --git a/flexynesis/config.py b/flexynesis/config.py
@@ -11,7 +11,7 @@
         Integer(32, 128, name='batch_size'),
         Categorical(epochs, name='epochs')
     ], 
-    'SVAE': [
+    'supervised_vae': [
         Integer(16, 128, name='latent_dim'),
         Integer(64, 512, name='hidden_dim'),
         Integer(8, 32, name='supervisor_hidden_dim'),