Source code for mlflow.entities.run_data

from mlflow.entities._mlflow_object import _MLflowObject
from mlflow.entities.metric import Metric
from mlflow.entities.param import Param
from mlflow.entities.run_tag import RunTag
from mlflow.protos.service_pb2 import Param as ProtoParam
from mlflow.protos.service_pb2 import RunData as ProtoRunData
from mlflow.protos.service_pb2 import RunTag as ProtoRunTag


[docs]class RunData(_MLflowObject): """ Run data (metrics and parameters). """ def __init__(self, metrics=None, params=None, tags=None): """Construct a new mlflow.entities.RunData instance. Args: metrics: List of mlflow.entities.Metric. params: List of mlflow.entities.Param. tags: List of mlflow.entities.RunTag. """ # Maintain the original list of metrics so that we can easily convert it back to # protobuf self._metric_objs = metrics or [] self._metrics = {metric.key: metric.value for metric in self._metric_objs} self._params = {param.key: param.value for param in (params or [])} self._tags = {tag.key: tag.value for tag in (tags or [])} @property def metrics(self): """ Dictionary of string key -> metric value for the current run. For each metric key, the metric value with the latest timestamp is returned. In case there are multiple values with the same latest timestamp, the maximum of these values is returned. """ return self._metrics @property def params(self): """Dictionary of param key (string) -> param value for the current run.""" return self._params @property def tags(self): """Dictionary of tag key (string) -> tag value for the current run.""" return self._tags def _add_metric(self, metric): self._metrics[metric.key] = metric.value self._metric_objs.append(metric) def _add_param(self, param): self._params[param.key] = param.value def _add_tag(self, tag): self._tags[tag.key] = tag.value
[docs] def to_proto(self): run_data = ProtoRunData() run_data.metrics.extend([m.to_proto() for m in self._metric_objs]) run_data.params.extend([ProtoParam(key=key, value=val) for key, val in self.params.items()]) run_data.tags.extend([ProtoRunTag(key=key, value=val) for key, val in self.tags.items()]) return run_data
[docs] def to_dictionary(self): return { "metrics": self.metrics, "params": self.params, "tags": self.tags, }
[docs] @classmethod def from_proto(cls, proto): run_data = cls() # iterate proto and add metrics, params, and tags for proto_metric in proto.metrics: run_data._add_metric(Metric.from_proto(proto_metric)) for proto_param in proto.params: run_data._add_param(Param.from_proto(proto_param)) for proto_tag in proto.tags: run_data._add_tag(RunTag.from_proto(proto_tag)) return run_data