⏪ random small defaults restored

RasmussenLab · Jun 4, 2024 · 23cb783 · 23cb783
1 parent 2cb9aa5
commit 23cb783
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 6 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
@@ -19,11 +19,11 @@ feature_names:
 
 model:
   num_hidden:
-    - 12
-  num_latent: 8 #150
+    - 1000
+  num_latent: 150
   beta: .0001
   dropout: .1
 
 training_loop:
   lr: 1e-4
-  num_epochs: 300
+  num_epochs: 40