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..76552e14 100644 --- 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