CODEWITHSUNDEEP
google-cloud-platformcomputer-visionautomlgoogle-cloud-automl
Alberto Tono
Alberto Tono

Reputation: 324

AutoML VISION Google SingleLabel Classification output TopK results

Currently AutoML Vision API is outputting a SingleLabel with the respective Score

For example:

I trained the model with 3 classes:

  1. A
  2. B
  3. C

Then when I am using Test & Use and I am uploading another image, I got only [CURRENT OUTPUT]

  1. Class A and 0.988437 / 0.99

Is there a way I can get this type of output with Top_K classes ( for example Top 3 (k=3) ) [DESIRED OUTPUT]

  1. Class A and 0.988437 / 0.99
  2. Class C and 0.3551 / 0.36
  3. Class B and 0.1201 / 0.12

Sorted based on their Score.

Thanks in Advance.

Upvotes: 0

Views: 40

Answers (1)

Ksign
Ksign

Reputation: 817

Single-label classification assigns a single label to each classified image and it returns only one predicted class.

Multi-label is more suited for your use case as it allows an image to be assigned multiple labels.

In the UI (which is what you seem to be using) you can specify the type of classification you want your custom model to perform when you create your dataset.

enter image description here

If, for any reason, you would like to have the option to get all/k predicted classes scores on the single-label classification, I suggest that you raise a Feature Request.

Upvotes: 0

Related Questions