ai.djl.modality

Class Classifications.Classification

java.lang.Object

ai.djl.modality.Classifications.Classification

Direct Known Subclasses:

DetectedObjects.DetectedObject

Enclosing class:

Classifications

public static class Classifications.Classification
extends java.lang.Object

A Classification stores the classification result for a single class on a single input.

Constructor Summary

Constructors

Constructor and Description
Classification(java.lang.String className, double probability) Constructs a single class result for a classification.

Method Summary

Modifier and Type	Method and Description
java.lang.String	getClassName() Returns the class name.
double	getProbability() Returns the probability.
java.lang.String	toString()

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, wait, wait, wait

Constructor Detail

Classification

public Classification(java.lang.String className,
                      double probability)

Constructs a single class result for a classification.

Parameters:

className - the class name of the result

probability - the probability of the result

Method Detail

getClassName

public java.lang.String getClassName()

Returns the class name.

Returns:

the class name

getProbability

public double getProbability()

Returns the probability.

Probability explains how accurately the classifier identified the target class.

Returns:

the probability

toString

public java.lang.String toString()

Overrides:

toString in class java.lang.Object