Rennan Haro
11/07/2023, 2:02 PMkedro-mlflow
is it possible to retrieve a model from S3 based on it’s name and stage, directly from a catalog entry? E.g., I’d like to get the production
version of the foobar
model. The .pkl
is saved as an artifact on S3.
I was thinking of building a custom dataset that calls the Mlflow REST API, get’s the path to the model with the given name and stage, and then downloads the artifact from S3, but I wonder if there is a simpler/better way of doing it.marrrcin
11/07/2023, 2:08 PMYolan Honoré-Rougé
11/07/2023, 2:34 PMMlflowModelRegistryDataSet
is the way to go ;)Rennan Haro
11/07/2023, 3:00 PM