Commit 4830457d authored by lucas_miranda's avatar lucas_miranda
Browse files

Changed defaults in deepof_experiments.smk

parent 8ae6ee69
Pipeline #92168 passed with stage
in 23 minutes and 29 seconds
......@@ -219,7 +219,7 @@ arena_dims = args.arena_dims
batch_size = args.batch_size
hypertun_trials = args.hpt_trials
encoding_size = args.encoding_size
exclude_bodyparts = tuple(args.exclude_bodyparts.split(","))
exclude_bodyparts = args.exclude_bodyparts.split(",")
gaussian_filter = args.gaussian_filter
hparams = args.hyperparameters
input_type = args.input_type
......
......@@ -91,7 +91,7 @@ rule explore_encoding_dimension_and_loss_function:
"--encoding-size {wildcards.encs} "
"--batch-size 256 "
"--window-size 24 "
"--window-step 24 "
"--window-step 6 "
"--exclude-bodyparts Tail_base,Tail_1,Tail_2,Tail_tip "
"--stability-check 3 "
"--output-path {outpath}dimension_and_loss_experiments"
......@@ -124,3 +124,8 @@ rule explore_phenotype_classification:
"--window-step 11 "
"--stability-check 3 "
"--output-path {outpath}pheno_classification_experiments"
"pipenv run python -m deepof.train_model --train-path {input.data_path} --val-num 5 --components {wildcards.k} --input-type coords --predictor 0 --phenotype-classifier 0 --variational True --loss {wildcards.loss} --kl-warmup 20 --mmd-warmup 20 --montecarlo-kl 10 --encoding-size {wildcards.encs} --batch-size 256 --window-size 24 --window-step 6 --exclude-bodyparts Tail_base,Tail_1,Tail_2,Tail_tip --stability-check 3 --output-path {outpath}dimension_and_loss_experiments"
\ No newline at end of file
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