Understanding Models

Last updated 8 days ago

Once you have trained a model, in addition to looking at the evaluation metrics and knowing how good in general your model is, another challenge is to understand the strengths and weaknesses of your model (on which kind of images it is good, on which kind of images it needs to improve).

To do so, you can select the model you are interested in as Current in the model listing page, by clicking on the checkbox, as seen on the screenshot below.

The model "v3" has been selected as the current model

Gallery Views

Starting from here, all gallery views in Studio will display prediction scores and will be ranked to help you understand what your model has learnt. The images are ranked from the ones that are the less representative of the concept you've selected to the ones that are the most representative of that concept (according to the current model).

Here is the kind of view that you will be able to get when clicking on a particular concept.

Image Views

You will also be able to visualize predictions on the Image views.

Classification & Tagging

For classification and tagging projects, you will get a score next to each concept corresponding to the confidence score of the neural network for the given concept.

Detection

For detection projects, you will be able to activate the prediction and visualize the bounding boxes proposed by the neural network. Each bounding box comes with a confidence score that is displayed in the navigation bar. To display those predictions, you need to click on the eye icon next to PREDICTIONS in the navigation bar.

In addition, you may want to filter the predictions by using their score. By default, only predictions above the optimal threshold are displayed. You can change that setting using the range control show below:

Score threshold controller

By changing this value, you will display more or less predicted boxes. This setting is defined concept-wise.