added loggers

This commit is contained in:
William Falcon
2020-01-16 17:24:03 -05:00
parent ae3dbf556d
commit 83a233e691
+9
View File
@@ -39,6 +39,15 @@ logger = getLogger(__name__)
class MLFlowLogger(LightningLoggerBase):
def __init__(self, experiment_name, tracking_uri=None, tags=None):
r"""
Logs using MLFlow
Args:
experiment_name (str): The name of the experiment
tracking_uri (str): where this should track
tags (dict): ?
"""
super().__init__()
self._mlflow_client = mlflow.tracking.MlflowClient(tracking_uri)
self.experiment_name = experiment_name