Adds output path option to train_model.py

deepof_experiments.smk

@@ -21,7 +21,8 @@ encodings = [2, 4, 6, 8, 12, 16]
 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",
+            "/u/lucasmir/DLC/DLC_autoencoders/DeepOF/deepof/logs/dimension_and_loss_experiments/trained_weights/\
+            GMVAE_loss={loss}_encoding={encs}_final_weights.h5",
             loss=losses,
             encs=encodings,
         ),
@@ -36,7 +37,11 @@ rule explore_encoding_dimension_and_loss_function:
             "dimension_and_loss_experiments/trained_weights/GMVAE_loss={loss}_encoding={encs}_final_weights.h5",
         ),
     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(
+        "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(
             outpath
         )