liuyu/qlib
1
0
Fork 0
mirror of https://github.com/microsoft/qlib.git synced 2026-07-12 15:26:54 +08:00
Code Activity

Support resuming recorder

Browse Source
This commit is contained in:
Jactus
2021-03-15 13:50:10 +08:00
parent 1d2b2f4f01
commit 734bb9ee3c
1 changed files with 6 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
qlib/workflow/exp.py
View File

@@ -180,10 +180,11 @@ class MLflowExperiment(Experiment):
def start(self, recorder_name=None): def start(self, recorder_name=None):
logger.info(f"Experiment {self.id} starts running ...") logger.info(f"Experiment {self.id} starts running ...")
# set up recorder # Get or create recorder
recorder = self.create_recorder(recorder_name) recorder = self._get_or_create_rec(recorder_name=recorder_name)
# Set up active recorder
self.active_recorder = recorder self.active_recorder = recorder
# start the recorder # Start the recorder
self.active_recorder.start_run() self.active_recorder.start_run()
return self.active_recorder return self.active_recorder
@@ -222,6 +223,8 @@ class MLflowExperiment(Experiment):
automatically create a new recorder based on the given id and name. automatically create a new recorder based on the given id and name.
""" """
try: try:
if recorder_id is None and recorder_name is None:
recorder_name = self._default_rec_name
return self._get_recorder(recorder_id=recorder_id, recorder_name=recorder_name), False return self._get_recorder(recorder_id=recorder_id, recorder_name=recorder_name), False
except ValueError: except ValueError:
if recorder_name is None: if recorder_name is None: