Added negative log likelihood of a Bernoulli distribution as reconstruction loss

deepof_experiments.smk

@@ -16,7 +16,7 @@ import os
 outpath = "/u/lucasmir/DLC/DLC_autoencoders/DeepOF/deepof/logs/"
 losses = ["ELBO", "MMD", "ELBO+MMD"]
 encodings = [2, 4, 6, 8, 10, 12, 14, 16]
-cluster_numbers = [i+1 for i in range(10)]
+cluster_numbers = [10]
 
 rule deepof_experiments:
     input:
@@ -29,33 +29,33 @@ rule deepof_experiments:
         ),
 
 
-rule coarse_hyperparameter_tuning:
-    input:
-        data_path="/u/lucasmir/DLC/DLC_models/deepof_single_topview/",
-    output:
-        trained_models=os.path.join(
-            outpath,
-            "coarse_hyperparameter_tuning/trained_weights/GMVAE_loss={loss}_encoding={encs}_run_1_final_weights.h5",
-        ),
-    shell:
-        "pipenv run python -m deepof.train_model "
-        "--train-path {input.data_path} "
-        "--val-num 25 "
-        "--components 10 "
-        "--input-type coords "
-        "--predictor 0 "
-        "--variational True "
-        "--loss {wildcards.loss} "
-        "--kl-warmup 20 "
-        "--mmd-warmup 20 "
-        "--encoding-size 2 "
-        "--batch-size 256 "
-        "--window-size 11 "
-        "--window-step 11 "
-        "--exclude-bodyparts Tail_base,Tail_1,Tail_2,Tail_tip,Spine_2 "
-        "--output-path {outpath}coarse_hyperparameter_tuning "
-        "--hyperparameter-tuning hyperband "
-        "--hpt-trials 3"
+# rule coarse_hyperparameter_tuning:
+#     input:
+#         data_path="/u/lucasmir/DLC/DLC_models/deepof_single_topview/",
+#     output:
+#         trained_models=os.path.join(
+#             outpath,
+#             "coarse_hyperparameter_tuning/trained_weights/GMVAE_loss={loss}_encoding={encs}_run_1_final_weights.h5",
+#         ),
+#     shell:
+#         "pipenv run python -m deepof.train_model "
+#         "--train-path {input.data_path} "
+#         "--val-num 25 "
+#         "--components 10 "
+#         "--input-type coords "
+#         "--predictor 0 "
+#         "--variational True "
+#         "--loss {wildcards.loss} "
+#         "--kl-warmup 20 "
+#         "--mmd-warmup 20 "
+#         "--encoding-size 2 "
+#         "--batch-size 256 "
+#         "--window-size 11 "
+#         "--window-step 11 "
+#         "--exclude-bodyparts Tail_base,Tail_1,Tail_2,Tail_tip,Spine_2 "
+#         "--output-path {outpath}coarse_hyperparameter_tuning "
+#         "--hyperparameter-tuning hyperband "
+#         "--hpt-trials 3"
 
 rule explore_encoding_dimension_and_loss_function:
     input: