Upload image predictions
Upload video predictions
Upload text predictions
Upload geospatial predictions
Upload HTML predictions
Overview
You will need to upload your model predictions to Labelbox to perform model error analysis, use model-assisted labeling, or set up active learning workflows. This page contains instructions for uploading model predictions to a model run, which is a container for the predictions, annotations, and metrics of a specific experiment in your ML model development cycle. You can include confidence scores and custom metrics when you upload your model predictions to a model run. However, given the predictions and annotations in a model run, Labelbox will automatically calculate some auto-generated metrics upon upload.
How to upload predictions to a model run
Step 1: Import data rows
To upload predictions in a model run, you’ll need to have a set of data rows to upload the predictions on. If you do not already have a set of data rows, you’ll need to import data rows in a dataset in Catalog. To learn how to import data rows via the Python SDK (Step 1), see this tutorial. To learn more about creating data rows, see Create a dataset in Catalog.Step 2: Create/select an ontology for your model predictions
When you create a Model - to upload a set of predictions - you’ll need to specify the ontology (also called taxonomy) that corresponds to the set of predictions. You may want to re-use an ontology that already exists in Labelbox - e.g. an ontology already used for a labeling project. Or, you may want to use an ontology for your model predictions that does not exist in Labelbox yet. In the latter case, you’ll need to create an ontology. To learn how to create an ontology via the Python SDK (Step 2), see this tutorial. To learn more about creating ontologies, see Create/modify ontologies.Step 3: Create an experiment and a model run
Before you can upload your predictions, you’ll need to make sure you have an experiment and a model run to contain them. You cannot simply upload predictions without specifying which Model and model run they’ll be associated with. An experiment is attached to an ontology: it defines the predictions that can be contained in it. An experiment can contain any number of model runs. Each model run corresponds to an experiment in your ML model development cycle. If you do not already have a Model and a model run, you will need to create a Model, then attach the ontology that fits your predictions, and finally create a model run. To learn how to set up a Model and model run via the Python SDK (step 3), see this tutorial. For more information, see experiments, Create a model run, Create a project.Step 4: Send a set of data rows to the model run
Now that you have your Model, model run, and ontology configured, you’ll need to send a set of data rows to the model run. To learn how to send data rows to a model run via the Python SDK (Step 4), see this tutorial.Step 5: Optionally, send annotations to the model run
After you have successfully configured your model run with the correct ontology and sent data rows to the model run, you can optionally send annotations to the model run. To learn how to send annotations to a model run via the Python SDK (Step 5), see this tutorial.Step 6: Create the predictions payload
Uploading confidence scores with each model predictions
Labelbox now supports specifying the confidence score associated with every model prediction. It is optional.Option A: Import predictions via Python Annotation types (recommended)
Option B: Import predictions via NDJSON
Image | Video | Text | Documents | Geospatial | Audio | Conversational text | |
---|---|---|---|---|---|---|---|
Bounding box | Payload | Payload | N/A | Payload | Payload | N/A | N/A |
Polygon | Payload | - | N/A | N/A | Payload | N/A | N/A |
Point | Payload | Payload | N/A | N/A | Payload | N/A | N/A |
Polyline | Payload | Payload | N/A | N/A | Payload | N/A | N/A |
Segmentation mask | Payload | Payload | N/A | N/A | - | N/A | N/A |
Text entity | N/A | N/A | Payload | Payload | N/A | N/A | Payload |
Classification - Radio | Payload | Payload | Payload | Payload | Payload | Payload | Payload |
Classification - Checklist | Payload | Payload | Payload | Payload | Payload | Payload | Payload |
Classification - Free-form text | Payload | - | Payload | Payload | Payload | Payload | Payload |
Relationship | Payload | - | Payload | Payload | - | N/A | Payload |