iicd/autodl-projects
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update models

Browse Source
This commit is contained in:
D-X-Y
2021-03-23 11:13:51 +00:00
parent 01397660de
commit 379b904203
7 changed files with 175 additions and 38 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
exps/trading/baselines.py
View File

@@ -15,6 +15,7 @@
# python exps/trading/baselines.py --alg TabNet #
# #
# python exps/trading/baselines.py --alg Transformer#
# python exps/trading/baselines.py --alg TSF-A #
#####################################################
import sys
import argparse
@@ -59,6 +60,7 @@ def retrieve_configs():
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"
alg2names["TSF-A"] = "workflow_config_transformer_basic_Alpha360.yaml"
# find the yaml paths
alg2paths = OrderedDict()