Hi friends my team and I have developed an `MLFlowDataSet` h

Question

Hi friends my team and I have developed an `MLFlowDataSet` 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

Jonas Kemper · Answer

think ``` def save self data gt 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 gt Any return mlflow sklearn load model mlflow model uri ```

Andrew Stewart · Answer

How s it relate differ from <https pypi org project kedro mlflow |https pypi org project kedro mlflow >

Andrew Stewart · Answer

Wonder if it might be a good addition to that plugin

Deepyaman Datta · Answer

Very late but agree that I m not sure how it would differ from what Kedro MLflow offers From a previous look through that plugin I think the datasets are implemented in a pretty top notch way