Commit dd013635 authored by lucas_miranda's avatar lucas_miranda
Browse files

Added negative log likelihood of a Bernoulli distribution as reconstruction loss

parent 42d242ce
Pipeline #88907 passed with stage
in 26 minutes and 2 seconds
......@@ -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:
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment