mlflow: delete all versions of a model
# Delete all versions of a model:
client = mlflow.tracking.MlflowClient()
model_name = "model_name"
# get the range manually
for i in range(35,43):
# convert first to archived stage else can't be deleted
client.transition_model_version_stage(
name=model_name,
version=i,
stage="archived"
)
# delete the version
client.delete_model_version(model_name, i)
Comments
Post a Comment