Updates

2021-03-17 03:51:48 +00:00
parent e04f17116d
commit 1ba1585f20
5 changed files with 86 additions and 7 deletions
--- a/exps/trading/baselines.py
+++ b/exps/trading/baselines.py
@@ -56,6 +56,7 @@ def retrieve_configs():
    alg2names["TabNet"] = "workflow_config_TabNet_Alpha360.yaml"
    alg2names["NAIVE-V1"] = "workflow_config_naive_v1_Alpha360.yaml"
    alg2names["NAIVE-V2"] = "workflow_config_naive_v2_Alpha360.yaml"
+    alg2names["Transformer"] = "workflow_config_transformer_Alpha360.yaml"

    # find the yaml paths
    alg2paths = OrderedDict()