⏪ random small defaults restored

RasmussenLab · Jun 4, 2024 · 0dc2a0f · 0dc2a0f
1 parent 2cb9aa5
commit 0dc2a0f
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 20 deletions.
diff --git a/tutorial/config/task/random_small__id_assoc_bayes.yaml b/tutorial/config/task/random_small__id_assoc_bayes.yaml
@@ -3,7 +3,7 @@ defaults:
 
 batch_size: 10
 
-num_refits: 40 # default is 40
+num_refits: 40
 
 target_dataset: random.small.drugs
 target_value: 1

diff --git a/tutorial/config/task/random_small__id_assoc_ttest.yaml b/tutorial/config/task/random_small__id_assoc_ttest.yaml
@@ -3,7 +3,7 @@ defaults:
 
 batch_size: 10 # number of samples per batch in training loop
 
-num_refits: 5 # 10 is the default  # number of times to refit (retrain) model
+num_refits: 10 # number of times to refit (retrain) model
 
 target_dataset: random.small.drugs # dataset to perturb
 target_value: 1 # value to change to
@@ -15,4 +15,4 @@ model: # model configuration
 
 training_loop: # training loop configuration
   lr: 1e-4 # learning rate
-  num_epochs: 40 # 40 is the default # number of epochs
+  num_epochs: 40 # number of epochs
diff --git a/tutorial/config/task/random_small__latent.yaml b/tutorial/config/task/random_small__latent.yaml
@@ -4,26 +4,15 @@ defaults:
 batch_size: 10
 
 feature_names:
-  - proteomics_1
-  - proteomics_2
-  - proteomics_3
-  - proteomics_4
-  - proteomics_5
-  - metagenomics_1
-  - metagenomics_4
-  - metagenomics_5
-  - metagenomics_10
-  #- drugs_20
-  #- metagenomics_174
-  #- proteomics_195
+  - drugs_20
+  - metagenomics_174
+  - proteomics_195
 
 model:
   num_hidden:
-    - 12
-  num_latent: 8 #150
-  beta: .0001
-  dropout: .1
+    - 1000
+  num_latent: 150
 
 training_loop:
   lr: 1e-4
-  num_epochs: 300
+  num_epochs: 40