Commit d8adb04f authored by lucas_miranda's avatar lucas_miranda
Browse files

Added extra branch to main autoencoder for rule_based prediction

parent cafff35c
Pipeline #98512 passed with stages
in 17 minutes and 54 seconds
...@@ -102,8 +102,8 @@ rule coarse_hyperparameter_tuning: ...@@ -102,8 +102,8 @@ rule coarse_hyperparameter_tuning:
"--rule-based-prediction {wildcards.rulesweight} " "--rule-based-prediction {wildcards.rulesweight} "
"--variational True " "--variational True "
"--loss {wildcards.loss} " "--loss {wildcards.loss} "
"--kl-warmup 5 " "--kl-warmup 30 "
"--mmd-warmup 5 " "--mmd-warmup 30 "
"--encoding-size {wildcards.enc} " "--encoding-size {wildcards.enc} "
"--batch-size 256 " "--batch-size 256 "
"--window-size 24 " "--window-size 24 "
...@@ -143,8 +143,8 @@ rule train_models: ...@@ -143,8 +143,8 @@ rule train_models:
"--variational True " "--variational True "
"--latent-reg {wildcards.latreg} " "--latent-reg {wildcards.latreg} "
"--loss {wildcards.loss} " "--loss {wildcards.loss} "
"--kl-warmup 5 " "--kl-warmup 30 "
"--mmd-warmup 5 " "--mmd-warmup 30 "
"--montecarlo-kl 10 " "--montecarlo-kl 10 "
"--encoding-size {wildcards.encs} " "--encoding-size {wildcards.encs} "
"--entropy-knn {wildcards.entknn} " "--entropy-knn {wildcards.entknn} "
......
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