Skip to content

Instantly share code, notes, and snippets.

@IMOKURI
Created October 18, 2021 03:42
Show Gist options
  • Save IMOKURI/219c35a289576b7c597c739bbd2b20f1 to your computer and use it in GitHub Desktop.
Save IMOKURI/219c35a289576b7c597c739bbd2b20f1 to your computer and use it in GitHub Desktop.
MLflow delete registered model
import mlflow
from mlflow.tracking import MlflowClient
mlflow.set_tracking_uri("http://127.0.0.1:5000")
client = MlflowClient()
# versions=[1, 2, 3]
# for version in versions:
# client.delete_model_version(name="sk-learn-random-forest-reg-model", version=version)
# Delete a registered model along with all its versions
client.delete_registered_model(name="resnet18")
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment