Add model saving for qrun and workflow example

microsoft · Feb 7, 2021 · c948385 · c948385
1 parent 07b905c
commit c948385
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/examples/workflow_by_code.py b/examples/workflow_by_code.py
@@ -112,6 +112,7 @@
     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
@@ -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"]: