From 23cb7838f5571df78e83f404ff09f51a06ee8bdb Mon Sep 17 00:00:00 2001 From: Henry Date: Tue, 4 Jun 2024 13:41:10 +0200 Subject: [PATCH] :rewind: random small defaults restored --- tutorial/config/task/random_small__id_assoc_bayes.yaml | 2 +- tutorial/config/task/random_small__id_assoc_ttest.yaml | 4 ++-- tutorial/config/task/random_small__latent.yaml | 6 +++--- 3 files changed, 6 insertions(+), 6 deletions(-) diff --git a/tutorial/config/task/random_small__id_assoc_bayes.yaml b/tutorial/config/task/random_small__id_assoc_bayes.yaml index be596e10..ad7e67f7 100644 --- 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 index 7ad0c65f..0a88162e 100644 --- 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 index e73e3d02..489f24ff 100644 --- 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