iicd/xautodl
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix bugs

Browse Source
This commit is contained in:
D-X-Y
2021-03-12 02:46:39 +00:00
parent 9f7eca0e58
commit 91a9e1553d
4 changed files with 7 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
exps/trading/baselines.py
View File

@@ -10,6 +10,7 @@
# python exps/trading/baselines.py --alg XGBoost #
# python exps/trading/baselines.py --alg LightGBM #
# python exps/trading/baselines.py --alg DoubleE #
# python exps/trading/baselines.py --alg TabNet #
#####################################################
import sys
import argparse
@@ -66,6 +67,7 @@ def retrieve_configs():
def main(xargs, exp_yaml):
assert Path(exp_yaml).exists(), "{:} does not exist.".format(exp_yaml)
pprint('Run {:}'.format(xargs.alg))
with open(exp_yaml) as fp:
config = yaml.safe_load(fp)
config = update_market(config, xargs.market)