/ examples / pyfunc / custom_code.py
custom_code.py
1  flower_classes = ["setosa", "versicolor", "virginica"]
2  
3  
4  def iris_classes(preds):
5      return [flower_classes[x] for x in preds]