Classification Unit

The classifier unit is used for inferring models that produce classes and confidence scores. This is a special case of the attribute model unit, where the output is a single class and confidence score.

Note

The ‘@classifier’ suffix in the unit name is an alias for the attribute model unit. Read more about it in the Attribute Model Unit section.

Example

- element: nvinfer@classifier
  name: classifier_model
  model:
    format: onnx
    remote:
      url: s3://savant-data/models/classifier_model/classifier_model.zip
      checksum_url: s3://savant-data/models/classifier_model/classifier_model.md5