diff --git a/deepof_experiments.smk b/deepof_experiments.smk index 5daeda1628a2802306f3f22e9a02a20824bbe7c0..db2c272fdf591ef9830bbe180b77f1e3afacba5a 100644 --- a/deepof_experiments.smk +++ b/deepof_experiments.smk @@ -22,7 +22,7 @@ rule deepof_experiments: input: expand( "/u/lucasmir/DLC/DLC_autoencoders/DeepOF/deepof/logs/dimension_and_loss_experiments/trained_weights/\ - GMVAE_loss={loss}_encoding={encs}_final_weights.h5", + GMVAE_loss={loss}_encoding={encs}_final_weights.h5", loss=losses, encs=encodings, ), @@ -38,10 +38,10 @@ rule explore_encoding_dimension_and_loss_function: ), shell: "pipenv run python -m deepof.train_model --train-path {input.data_path} --val-num 10 --components 10 \ - --input-type coords --predictor 0 --variational True --loss {wildcards.loss} --kl-warmup 20 --mmd-warmup 20 \ - --encoding-size {wildcards.encs} --batch-size 256 --window-size 11 --window-step 11 --exclude-bodyparts \ - Tail_base,Tail_1,Tail_2,Tail_tip,Spine_2 --stability-check 3 --out-path \ - {}dimension_and_loss_experiments/trained_weights".format( + --input-type coords --predictor 0 --variational True --loss {wildcards.loss} --kl-warmup 20 --mmd-warmup 20 \ + --encoding-size {wildcards.encs} --batch-size 256 --window-size 11 --window-step 11 --exclude-bodyparts \ + Tail_base,Tail_1,Tail_2,Tail_tip,Spine_2 --stability-check 3 --out-path \ + {}dimension_and_loss_experiments/trained_weights".format( outpath )