Skip to content

Instantly share code, notes, and snippets.

@HemachandranD
Created March 8, 2024 18:00
Show Gist options
  • Save HemachandranD/2f094e442e6a33d10902afef9532b7fd to your computer and use it in GitHub Desktop.
Save HemachandranD/2f094e442e6a33d10902afef9532b7fd to your computer and use it in GitHub Desktop.
# Log the trained model with MLflow and package it with feature lookup information.
# fs.log_model(
# model,
# artifact_path="model_packaged",
# flavor=mlflow.lightgbm,
# training_set=training_set,
# registered_model_name=model_name,
# )
# Log the trained model with MLflow.
autolog_run = mlflow.last_active_run()
model_uri = "runs:/{}/model".format(autolog_run.info.run_id)
mlflow.register_model(model_uri, model_name)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment