Skip to content
Snippets Groups Projects
Commit 56ef649b authored by Lucas Miranda's avatar Lucas Miranda
Browse files

Adds output path option to train_model.py

parent f221653f
Branches
No related tags found
No related merge requests found
...@@ -22,7 +22,7 @@ rule deepof_experiments: ...@@ -22,7 +22,7 @@ rule deepof_experiments:
input: input:
expand( expand(
"/u/lucasmir/DLC/DLC_autoencoders/DeepOF/deepof/logs/dimension_and_loss_experiments/trained_weights/" "/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}_run_1_final_weights.h5",
loss=losses, loss=losses,
encs=encodings, encs=encodings,
), ),
...@@ -34,7 +34,7 @@ rule explore_encoding_dimension_and_loss_function: ...@@ -34,7 +34,7 @@ rule explore_encoding_dimension_and_loss_function:
output: output:
trained_models=os.path.join( trained_models=os.path.join(
outpath, outpath,
"dimension_and_loss_experiments/trained_weights/GMVAE_loss={loss}_encoding={encs}_final_weights.h5", "dimension_and_loss_experiments/trained_weights/GMVAE_loss={loss}_encoding={encs}_run_1_final_weights.h5",
), ),
shell: shell:
"pipenv run python -m deepof.train_model --train-path {input.data_path} --val-num 10 --components 10 \ "pipenv run python -m deepof.train_model --train-path {input.data_path} --val-num 10 --components 10 \
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment