Jonas Kemper
09/16/2022, 9:56 AMMLFlowDataSet
handler because we wanted to use it as our model registry. It's just a thin wrapper around https://www.mlflow.org/docs/latest/python_api/mlflow.sklearn.html . Could this potentially be interesting for others as well and might be a worthwhile contribution?def _save(self, data) -> None:
mlflow.set_tracking_uri(self.tracking_uri)
model_info = mlflow.sklearn.log_model(
sk_model=data,
artifact_path=self.artifact_path,
registered_model_name=self.registered_model_name,
)
and
def _load(self) -> Any:
[...]
return mlflow.sklearn.load_model(mlflow_model_uri)
Andrew Stewart
09/16/2022, 3:19 PMDeepyaman Datta
10/01/2022, 7:30 PM