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:
"--rule-based-prediction {wildcards.rulesweight} "
"--variational True "
"--loss {wildcards.loss} "
"--kl-warmup 5 "
"--mmd-warmup 5 "
"--kl-warmup 30 "
"--mmd-warmup 30 "
"--encoding-size {wildcards.enc} "
"--batch-size 256 "
"--window-size 24 "
......@@ -143,8 +143,8 @@ rule train_models:
"--variational True "
"--latent-reg {wildcards.latreg} "
"--loss {wildcards.loss} "
"--kl-warmup 5 "
"--mmd-warmup 5 "
"--kl-warmup 30 "
"--mmd-warmup 30 "
"--montecarlo-kl 10 "
"--encoding-size {wildcards.encs} "
"--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