diff --git a/docs/deployments/onnx.md b/docs/deployments/onnx.md index 9cae26ad71..c4b02d6d31 100644 --- a/docs/deployments/onnx.md +++ b/docs/deployments/onnx.md @@ -86,7 +86,7 @@ class ONNXPredictor: ```python import numpy as np -labels = ["iris-setosa", "iris-versicolor", "iris-virginica"] +labels = ["setosa", "versicolor", "virginica"] class ONNXPredictor: diff --git a/docs/deployments/python.md b/docs/deployments/python.md index 518843556f..d344a0b422 100644 --- a/docs/deployments/python.md +++ b/docs/deployments/python.md @@ -88,7 +88,7 @@ class PythonPredictor: import boto3 from my_model import IrisNet -labels = ["iris-setosa", "iris-versicolor", "iris-virginica"] +labels = ["setosa", "versicolor", "virginica"] class PythonPredictor: def __init__(self, config): diff --git a/docs/deployments/tensorflow.md b/docs/deployments/tensorflow.md index 53fa491027..55d41b52c4 100644 --- a/docs/deployments/tensorflow.md +++ b/docs/deployments/tensorflow.md @@ -85,7 +85,7 @@ class TensorFlowPredictor: ## Example ```python -labels = ["iris-setosa", "iris-versicolor", "iris-virginica"] +labels = ["setosa", "versicolor", "virginica"] class TensorFlowPredictor: