From c948385e769fd23bbd6153ea1d0a94b26a01e6f9 Mon Sep 17 00:00:00 2001 From: Jactus Date: Sun, 7 Feb 2021 10:06:33 +0800 Subject: [PATCH] Add model saving for qrun and workflow example --- examples/workflow_by_code.py | 1 + qlib/model/trainer.py | 1 + 2 files changed, 2 insertions(+) diff --git a/examples/workflow_by_code.py b/examples/workflow_by_code.py index 4ef5d9b16..d5dab8917 100644 --- a/examples/workflow_by_code.py +++ b/examples/workflow_by_code.py @@ -112,6 +112,7 @@ if __name__ == "__main__": with R.start(experiment_name="workflow"): R.log_params(**flatten_dict(task)) model.fit(dataset) + R.save_objects(**{"params.pkl": model}) # prediction recorder = R.get_recorder() diff --git a/qlib/model/trainer.py b/qlib/model/trainer.py index 8f128d382..f0bc0b780 100644 --- a/qlib/model/trainer.py +++ b/qlib/model/trainer.py @@ -26,6 +26,7 @@ def task_train(task_config: dict, experiment_name): R.log_params(**flatten_dict(task_config)) model.fit(dataset) recorder = R.get_recorder() + R.save_objects(**{"params.pkl": model}) # generate records: prediction, backtest, and analysis for record in task_config["record"]: