Skip to content

Instantly share code, notes, and snippets.

@Steboss89

Steboss89/register_model.py

Created September 18, 2022 20:58
Show Gist options
  • Save Steboss89/cbc2202a1bf9f26998c930f248685229 to your computer and use it in GitHub Desktop.
Save Steboss89/cbc2202a1bf9f26998c930f248685229 to your computer and use it in GitHub Desktop.
Download ZIP
Register a model to MLflow registry
Raw
register_model.py
import mlflow
run_id= "ade1040e1fd94a74b1da5ec0583fba63"
artifact_name = "model"
model_name = "NAME OF YOU MODEL"
mlflow.register_model(f"runs:/{run_id}/{artifact_name}", model_name)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment