Model serving (TF Serving) in TensorFlow

Introduction

Model serving lets you use a trained machine learning model to make predictions anytime. TensorFlow Serving helps you do this easily and fast.

You want to let a website or app get predictions from your model.

You need to update your model without stopping the service.

You want to handle many prediction requests quickly.

You want to keep your model safe and separate from your app code.

Syntax

TensorFlow

tensorflow_model_server --rest_api_port=8501 --model_name=my_model --model_base_path=/path/to/my_model

This command starts TensorFlow Serving with REST API on port 8501.

Replace my_model and /path/to/my_model with your model's name and folder path.

Examples

Serve the MNIST model on REST API port 8501.

TensorFlow

tensorflow_model_server --rest_api_port=8501 --model_name=mnist --model_base_path=/models/mnist

Serve the Iris model using gRPC on port 8500 (default for gRPC).

TensorFlow

tensorflow_model_server --port=8500 --model_name=iris --model_base_path=/models/iris

Sample Model

This Python code sends data to a TensorFlow model served with TF Serving and prints the prediction result.

TensorFlow

import requests
import json

# URL of the TensorFlow Serving REST API
url = 'http://localhost:8501/v1/models/my_model:predict'

# Example input data for prediction
input_data = {
    "instances": [[5.1, 3.5, 1.4, 0.2]]  # Example features
}

# Send POST request to the model server
response = requests.post(url, data=json.dumps(input_data), headers={'Content-Type': 'application/json'})

# Print the prediction result
print('Prediction response:', response.json())

Important Notes

Make sure TensorFlow Serving is running and the model path is correct before sending requests.

You can use REST API or gRPC to communicate with TensorFlow Serving.

Model updates can be done by adding new versions in the model folder without stopping the server.

Summary

TensorFlow Serving helps you deploy ML models for real-time predictions.

You start the server with a command pointing to your saved model folder.

Clients send data to the server via REST or gRPC to get predictions.

Practice

1. What is the main purpose of TensorFlow Serving in machine learning?

easy

A. To clean and preprocess data before training

B. To train new machine learning models

C. To visualize model training progress

D. To deploy trained models for real-time predictions

Solution

1. Step 1: Understand TensorFlow Serving's role

   TensorFlow Serving is designed to deploy models so they can be used to make predictions quickly.

2. Step 2: Differentiate from training and preprocessing

   Training and preprocessing happen before serving; TensorFlow Serving focuses on using the trained model.

3. Final Answer:

   To deploy trained models for real-time predictions -> Option D

4. Quick Check:

   Serving = Deploy models for predictions [OK]

Hint: Serving means deploying models to answer queries fast [OK]

Common Mistakes:

• Confusing serving with training
• Thinking it preprocesses data
• Assuming it visualizes training

2. Which command correctly starts TensorFlow Serving to serve a model saved at /models/my_model?

easy

A. tensorflow_model_server --model_base_path=/models/my_model --rest_api_port=8501

B. tensorflow_serve --model_path=/models/my_model --port=8501

C. serve_tf_model --model_dir=/models/my_model --grpc_port=8501

D. start_tf_server --model=/models/my_model --api_port=8501

Solution

1. Step 1: Identify the official TensorFlow Serving command

   The official command is tensorflow_model_server with flags like --model_base_path and --rest_api_port.

2. Step 2: Match the correct flags and path

   tensorflow_model_server --model_base_path=/models/my_model --rest_api_port=8501 uses correct command and flags pointing to /models/my_model and port 8501.

3. Final Answer:

   tensorflow_model_server --model_base_path=/models/my_model --rest_api_port=8501 -> Option A

4. Quick Check:

   Correct command and flags = tensorflow_model_server --model_base_path=/models/my_model --rest_api_port=8501 [OK]

Hint: Look for 'tensorflow_model_server' and '--model_base_path' flags [OK]

Common Mistakes:

• Using wrong command name
• Incorrect flag names
• Wrong model path flag

3. Given this Python code snippet sending data to TensorFlow Serving REST API, what is the expected output if the model predicts class 1 with 0.85 confidence?

import requests
import json

url = 'http://localhost:8501/v1/models/my_model:predict'
data = {"instances": [[5.1, 3.5, 1.4, 0.2]]}
response = requests.post(url, json=data)
print(response.json())

medium

A. {"predictions": [[1, 0.85]]}

B. {"predictions": [[1]]}

C. {"predictions": [[0.85, 1]]}

D. {"error": "Model not found"}

Solution

1. Step 1: Understand the REST API response format

   TensorFlow Serving returns predictions as a list of outputs matching the model's output shape. For classification, usually a class label or probabilities.

2. Step 2: Match the expected output for class prediction

   If the model predicts class 1 with confidence 0.85, the output is typically the class label only, e.g., [1].

3. Final Answer:

   {"predictions": [[1]]} -> Option B

4. Quick Check:

   Prediction output = class label list [OK]

Hint: REST API returns predictions list matching model output shape [OK]

Common Mistakes:

• Confusing confidence with class label
• Expecting confidence and label together
• Ignoring model output format

4. You started TensorFlow Serving with this command:
tensorflow_model_server --model_base_path=/models/my_model --rest_api_port=8501
But when you send a request, you get a 404 error. What is the most likely cause?

medium

A. The model folder path is incorrect or missing the version subfolder

B. The REST API port 8501 is blocked by firewall

C. The client request JSON format is wrong

D. TensorFlow Serving does not support REST API

Solution

1. Step 1: Check model folder structure requirement

   TensorFlow Serving expects the model folder to contain version subfolders like 1/, 2/ with saved model files inside.

2. Step 2: Understand 404 error meaning

   A 404 error means the server cannot find the requested model, often due to missing or wrong model path or version folder.

3. Final Answer:

   The model folder path is incorrect or missing the version subfolder -> Option A

4. Quick Check:

   404 error = model path/version issue [OK]

Hint: Model folder must have version subfolders like '1/' [OK]

Common Mistakes:

• Ignoring model folder versioning
• Assuming REST API unsupported
• Blaming firewall without checking model path

5. You want to serve two different versions of a model simultaneously using TensorFlow Serving. Which setup allows clients to request a specific version?

hard

A. Rename model folders to different names and serve them under separate model names

B. Run two separate TensorFlow Serving servers on different ports, each serving one version

C. Place version folders (e.g., 1/, 2/) inside the model base path and use the version parameter in the request URL

D. TensorFlow Serving does not support serving multiple versions simultaneously

Solution

1. Step 1: Understand versioning in TensorFlow Serving

   TensorFlow Serving supports multiple versions by placing them as subfolders inside the model base path, e.g., 1/, 2/.

2. Step 2: How clients request specific versions

   Clients can specify the version in the request URL to get predictions from a particular version.

3. Final Answer:

   Place version folders (e.g., 1/, 2/) inside the model base path and use the version parameter in the request URL -> Option C

4. Quick Check:

   Version folders + version param = multiple versions [OK]

Hint: Use version folders and version param in request URL [OK]

Common Mistakes:

• Thinking separate servers are needed
• Renaming folders instead of versioning
• Believing multiple versions unsupported