From 25fc96b787db54bf0bd127aeb923c6b90917297f Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Ricardo=20Hern=C3=A1ndez=20Medina?=
 <ricardo.hernandez.medina@sund.ku.dk>
Date: Tue, 4 Jun 2024 14:46:04 +0200
Subject: [PATCH] :wrench: Add default KS config

---
 .../conf/task/identify_associations_ks.yaml   | 26 +++++++++++++++++++
 .../task/random_continuous__id_assoc_ks.yaml  | 13 ----------
 2 files changed, 26 insertions(+), 13 deletions(-)
 create mode 100644 src/move/conf/task/identify_associations_ks.yaml

diff --git a/src/move/conf/task/identify_associations_ks.yaml b/src/move/conf/task/identify_associations_ks.yaml
new file mode 100644
index 00000000..3c2e23af
--- /dev/null
+++ 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
index 1eb1b3e5..f825c7b6 100644
--- 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