Files
mlflow--mlflow/examples/pyfunc/custom_code.py
T
2026-07-13 13:22:34 +08:00

6 lines
128 B
Python

flower_classes = ["setosa", "versicolor", "virginica"]
def iris_classes(preds):
return [flower_classes[x] for x in preds]