from mlflow.entities._mlflow_object import _MlflowObject
from mlflow.protos.service_pb2 import ModelOutput
[docs]class LoggedModelOutput(_MlflowObject):
"""ModelOutput object associated with a Run."""
def __init__(self, model_id: str, step: int) -> None:
self._model_id = model_id
self._step = step
def __eq__(self, other: _MlflowObject) -> bool:
if type(other) is type(self):
return self.__dict__ == other.__dict__
return False
@property
def model_id(self) -> str:
"""Model ID"""
return self._model_id
@property
def step(self) -> str:
"""Step at which the model was logged"""
return self._step
[docs] def to_proto(self):
return ModelOutput(model_id=self.model_id, step=self.step)
[docs] @classmethod
def from_proto(cls, proto):
return cls(proto.model_id, proto.step)
RetroSearch is an open source project built by @garambo | Open a GitHub Issue
Search and Browse the WWW like it's 1997 | Search results from DuckDuckGo
HTML:
3.2
| Encoding:
UTF-8
| Version:
0.7.4