Skip to content

Instantly share code, notes, and snippets.

@awaiskaleem

awaiskaleem/load_model.py

Created July 14, 2020 19:46
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save awaiskaleem/b380468bddd67deb4ac81ba77deceb7b to your computer and use it in GitHub Desktop.
Save awaiskaleem/b380468bddd67deb4ac81ba77deceb7b to your computer and use it in GitHub Desktop.
Download ZIP
Raw
load_model.py
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