Skip to content

Instantly share code, notes, and snippets.

Embed
What would you like to do?
def load_model(run_id, model_name):
model = mlflow.get_run(run_id).info.artifact_uri+"/model_name/sparkml"
return PipelineModel.load(model)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment