🔧 Add default KS config

RasmussenLab · Jun 4, 2024 · 25fc96b · 25fc96b
1 parent 0dc2a0f
commit 25fc96b
Show file tree

Hide file tree

Showing 2 changed files with 26 additions and 13 deletions.
diff --git a/src/move/conf/task/identify_associations_ks.yaml b/src/move/conf/task/identify_associations_ks.yaml
@@ -0,0 +1,26 @@
+defaults:
+  - identify_associations_ks_schema
+
+model:
+  categorical_weights: ${weights:${data.categorical_inputs}}
+  continuous_weights: ${weights:${data.continuous_inputs}}
+  num_hidden:
+    - 100
+  num_latent: 50
+  beta: 0.1
+  dropout: 0.1
+  cuda: false
+
+training_loop:
+  lr: 1e-4
+  num_epochs: 200
+  batch_dilation_steps:
+    - 50
+    - 100
+    - 150
+  kld_warmup_steps:
+    - 15
+    - 20
+    - 25
+  early_stopping: false
+  patience: 0
diff --git a/tutorial/config/task/random_continuous__id_assoc_ks.yaml b/tutorial/config/task/random_continuous__id_assoc_ks.yaml
@@ -11,28 +11,15 @@ target_value: plus_std
 save_refits: True
 
 model:
-  categorical_weights: ${weights:${data.categorical_inputs}}
-  continuous_weights: ${weights:${data.continuous_inputs}}
   num_hidden:
     - 100
   num_latent: 50
   beta: 0.1
   dropout: 0.1
-  cuda: false
 
 training_loop:
   lr: 1e-4
   num_epochs: 200
-  batch_dilation_steps:
-    - 50
-    - 100
-    - 150
-  kld_warmup_steps:
-    - 15
-    - 20
-    - 25
-  early_stopping: false
-  patience: 0
 
 perturbed_feature_names:
   - proteomics_3