liuyu/qlib
1
0
Fork 0
mirror of https://github.com/microsoft/qlib.git synced 2026-07-06 20:41:09 +08:00
Code Activity

Modify cli

Browse Source
This commit is contained in:
Jactus
2020-11-20 10:59:37 +08:00
parent 0f433571f6
commit 547697ddc6
6 changed files with 16 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
qlib/workflow/cli.py
View File

@@ -14,7 +14,7 @@ from qlib.workflow import R
from qlib.workflow.record_temp import SignalRecord
# worflow handler function
def workflow(config_path):
def workflow(config_path, experiment_name="workflow"):
with open(config_path) as fp:
config = yaml.load(fp, Loader=yaml.Loader)
@@ -26,12 +26,13 @@ def workflow(config_path):
dataset = init_instance_by_config(config.get("task")["dataset"])
# start exp
with R.start(experiment_name="workflow"):
R.log_paramters(**flatten_dict(task))
with R.start(experiment_name=experiment_name):
# train model
R.log_params(**flatten_dict(config.get("task")))
model.fit(dataset)
recorder = R.get_recorder()
# generate records
# generate records: prediction, backtest, and analysis
for record in config.get("task")["record"]:
if record["class"] == SignalRecord.__name__:
srconf = {"model": model, "dataset": dataset, "recorder": recorder}