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

Format with black

Browse Source
This commit is contained in:
Jactus
2020-10-29 13:22:49 +08:00
parent 490dbd908b
commit da9d1c8ac6
20 changed files with 290 additions and 251 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
qlib/workflow/exp.py
View File

@@ -4,10 +4,12 @@
import mlflow
from pathlib import Path
class Experiment:
"""
Thie is the `Experiment` class for each experiment being run. The API is designed
Thie is the `Experiment` class for each experiment being run. The API is designed
"""
def __init__(self):
self.name = None
self.id = None
@@ -39,9 +41,10 @@ class MLflowExperiment(Experiment):
"""
Use mlflow to implement Experiment.
"""
def search_records(self, **kwargs):
filter_string = '' if kwargs.get('filter_string') is None else kwargs.get('filter_string')
run_view_type = 1 if kwargs.get('run_view_type') is None else kwargs.get('run_view_type')
max_results = 100000 if kwargs.get('max_results') is None else kwargs.get('max_results')
order_by = kwargs.get('order_by')
return mlflow.search_runs([self.experiment_id], filter_string, run_view_type, max_results, order_by)
filter_string = "" if kwargs.get("filter_string") is None else kwargs.get("filter_string")
run_view_type = 1 if kwargs.get("run_view_type") is None else kwargs.get("run_view_type")
max_results = 100000 if kwargs.get("max_results") is None else kwargs.get("max_results")
order_by = kwargs.get("order_by")
return mlflow.search_runs([self.experiment_id], filter_string, run_view_type, max_results, order_by)