1
0
mirror of https://github.com/microsoft/qlib.git synced 2026-07-12 23:36:54 +08:00

Update run_all_model

This commit is contained in:
Jactus
2020-12-02 17:55:11 +08:00
parent 745b93138d
commit 91c3dfddf5

View File

@@ -15,6 +15,7 @@ import traceback
import functools import functools
import statistics import statistics
import subprocess import subprocess
from datetime import datetime
from pathlib import Path from pathlib import Path
from operator import xor from operator import xor
from pprint import pprint from pprint import pprint
@@ -45,8 +46,6 @@ if not exists_qlib_data(provider_uri):
GetData().qlib_data(target_dir=provider_uri, region=REG_CN) GetData().qlib_data(target_dir=provider_uri, region=REG_CN)
qlib.init(provider_uri=provider_uri, region=REG_CN, exp_manager=exp_manager) qlib.init(provider_uri=provider_uri, region=REG_CN, exp_manager=exp_manager)
if os.path.isdir(exp_path):
shutil.rmtree(exp_path)
# decorator to check the arguments # decorator to check the arguments
def only_allow_defined_args(function_to_decorate): def only_allow_defined_args(function_to_decorate):
@@ -291,7 +290,8 @@ def run(times=1, models=None, exclude=False):
sys.stderr.write(f"Here are some of the errors of the models...\n") sys.stderr.write(f"Here are some of the errors of the models...\n")
pprint(errors) pprint(errors)
sys.stderr.write("\n") sys.stderr.write("\n")
# move results folder
shutil.move(exp_path, exp_path + f"_{datetime.now().strftime("%Y-%m-%d_%H:%M:%S")}")
if __name__ == "__main__": if __name__ == "__main__":
fire.Fire(run) # run all the model fire.Fire(run) # run all the model