register_model
Registers a machine learning model into the MLflow model registry, creating the registered model if it does not exist.
Instructions
Register a model into the model registry. Creates the registered model if it doesn't exist.
Args: model_name: Name for the registered model. model_uri: URI of the model to register. Supports: - LoggedModel: 'models:/m-abc123' - Run artifact: 'runs:/run_id/artifact_path' tags: Optional dict of tags to set on the model version.
Examples: register_model("btc-classifier", "models:/m-abc123") register_model("btc-classifier", "runs:/abc123/model", tags={"framework": "lightgbm"})
Input Schema
| Name | Required | Description | Default |
|---|---|---|---|
| model_name | Yes | ||
| model_uri | Yes | ||
| tags | No |