Make Visual AI predictions via the API¶

This tutorial outlines how to make predictions on Visual AI projects with API calls. To complete this tutorial, you must have trained and deployed a visual AI model.

Takeaways¶

This tutorial shows how to:

• Configure scripting code for making batch predictions via the API
• Make an API call to get batch predictions for a visual AI model
• Format images to a base64 format

Predictions workflow¶

1. Prepare your data for Visual AI. Before making predictions, convert the images you want to score to base64 format (the standard format for handling images in API calls). Note that when the model returns prediction results, images will return in base64 format. To convert data, use DataRobot's Python package, described in the guide Preparing binary data for predictions.

2. After training and deploying a Visual AI model, navigate to the deployment and access the Predictions > Prediction API tab. This tab provides the scripting code used to make predictions via the API.

3. To configure the scripting code, select Batch as the prediction type and API Client as the interface type.

   

4. Copy the code and save it as a Python script (e.g., datarobot-predict.py). You can edit the script to incorporate additional steps. For example, add the passthrough_columns_set argument to BatchPredictionJob if you would like to include columns from the input file (e.g., image_id) to the output file.

   

5. Using the scripting code from step two and a base64-converted image file (InputDataConverted.csv), make an API call to get predictions from the deployed model:

   python datarobot-predict.py InputDataConverted.csv Predictions.csv

6. Access the output file (Predictions.csv) to view prediction results.

Learn More¶

Documentation¶

• Visual AI overview
• Making predictions with Visual AI
• Prediction API overview